Python/fitotrack-logger
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

extrapolate owner from filename

Browse Source
This commit is contained in:
Riccardo Berto 2021-02-24 17:20:27 +01:00
parent a141403536
commit 533fb098e3
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
main.py
View File

@ -58,10 +58,12 @@ def get_gpx_files_from_mail():
mail.logout() mail.logout()
def process_gpx_files(tx: Connection, owner: str): def process_gpx_files(tx: Connection):
for filepath in glob('gpx_files/*.gpx'): for filepath in glob('gpx_files/*.gpx'):
filename = os.path.split(filepath)[-1] owner = os.path.split(filepath)[-1].split('_workout-')[0]
if list(db.execute(text('select exists(select from training where filename = :filename)'), dict(filename=filename)))[0][0]: filename = f'workout-{os.path.split(filepath)[-1].split("_workout-")[1]}'
if list(db.execute(text('select exists(select from training where owner = :owner and filename = :filename)'),
dict(owner=owner, filename=filename,),),)[0][0]:
continue continue
with open(filepath) as f: with open(filepath) as f:
gpx_file = gpxpy.parse(f) gpx_file = gpxpy.parse(f)
@ -95,15 +97,15 @@ def process_gpx_files(tx: Connection, owner: str):
elevation=point.elevation,),) elevation=point.elevation,),)
def main(owner: str): def main():
init_database() init_database()
get_gpx_files_from_mail() get_gpx_files_from_mail()
db.transaction(process_gpx_files, owner) db.transaction(process_gpx_files)
if __name__ == '__main__': if __name__ == '__main__':
try: try:
main(sys.argv[1]) main()
except IndexError: except IndexError:
print('Run the script with "python main.py OWNER_NAME"') print('Run the script with "python main.py OWNER_NAME"')
except (KeyboardInterrupt, EOFError): except (KeyboardInterrupt, EOFError):