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

enclose db ops in a transaction

Browse Source
This commit is contained in:
Riccardo Berto 2021-03-01 15:14:05 +01:00
parent 841fac5634
commit 228a4e64f4
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
main.py
View File

@ -72,7 +72,7 @@ def process_gpx_files(tx: Connection):
owner = os.path.split(filepath)[-1].split('_workout-')[0]
filename = f'workout-{os.path.split(filepath)[-1].split("_workout-")[1]}'
print(f'Processing {filename}')
if list(db.execute(text('select exists(select from training where owner = :owner and filename = :filename)'),
if list(tx.execute(text('select exists(select from training where owner = :owner and filename = :filename)'),
dict(owner=owner, filename=filename,),),)[0][0]:
os.remove(filepath)
continue
@ -80,7 +80,7 @@ def process_gpx_files(tx: Connection):
gpx_file = gpxpy.parse(f)
if gpx_file.creator != 'FitoTrack':
raise ValueError('gpx file not generated by the FitoTrack app')
training_id = list(db.execute(
training_id = list(tx.execute(
text("""
insert into training(owner, filename, type, description, moving_time, stopped_time, moving_distance, stopped_distance) values
(:owner, :filename, :type, :description, :moving_time, :stopped_time, :moving_distance, :stopped_distance) returning id
@ -97,7 +97,7 @@ def process_gpx_files(tx: Connection):
for track in gpx_file.tracks:
for segment in track.segments:
for point in segment.points:
db.execute(text("""
tx.execute(text("""
insert into training_data(training_id, t, geog, speed, elevation)
values (:training_id, :t, :geog, :speed, :elevation)
"""),