meteortools.ukmondb.ECSVhandler

 1# Copyright (C) 2018-2023 Mark McIntyre
 2
 3import requests
 4import os
 5
 6
 7def getECSVs(stationID, dateStr, savefiles=False, outdir='.'):
 8    """
 9    Retrieve a detection in ECSV format for the specified date  
10
11    Arguments:  
12        stationID:  [str] RMS Station ID code  
13        dateStr:    [str] Date/time to retrieve for in ISO1601 format   
14                          eg 2021-07-17T02:41:05.05  
15    
16    Keyword Arguments:  
17        saveFiles:  [bool] save to file, or print to screen. Default False  
18        outdir:     [str] path to save files into. Default '.'  
19    """
20    apiurl='https://api.ukmeteors.co.uk/getecsv?stat={}&dt={}'
21    res = requests.get(apiurl.format(stationID, dateStr))
22    ecsvlines=''
23    if res.status_code == 200:
24        rawdata=res.text.strip()
25        if len(rawdata) > 10:
26            ecsvlines=rawdata.split('\n') # convert the raw data into a python list
27            if savefiles is True:
28                os.makedirs(outdir, exist_ok=True)
29                fnamebase = dateStr.replace(':','_').replace('.','_') # create an output filename
30                j=0
31                outf = False
32                for li in ecsvlines:
33                    if 'issue getting data' in li:
34                        print(li)
35                        return li
36                    if '# %ECSV' in li:
37                        if outf is not False:
38                            outf.close()
39                        j=j+1
40                        fname = fnamebase + f'_ukmda_{stationID}_M{j:03d}.ecsv'
41                        outf = open(os.path.join(outdir, fname), 'w')
42                        print('saving to ', os.path.join(outdir,fname))
43                    if outf:
44                        outf.write(f'{li}\n')
45                    else:
46                        print('no ECSV marker found in data')
47        else:
48            print('no error, but no data returned')
49    else:
50        print(res.status_code)
51    return ecsvlines