marcel/pylot
3
0
Fork 0
Code Issues 7 Pull Requests 1 Releases Wiki Activity

[minor] make safety copy of existing deleted picks file before overwriting

Browse Source
This commit is contained in:
Marcel Paffrath 2019-06-24 15:02:07 +02:00
parent 6caa3ee9ce
commit fe73d2e054
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
PyLoT.py
View File

@ -27,6 +27,7 @@ import argparse
import matplotlib import matplotlib
import os import os
import platform import platform
import shutil
import sys import sys
import json import json
@ -2768,7 +2769,13 @@ class MainWindow(QMainWindow):
def dump_deleted_picks(self, event_path): def dump_deleted_picks(self, event_path):
''' Save deleted picks to json file for event in event_path. Load old file before and merge''' ''' Save deleted picks to json file for event in event_path. Load old file before and merge'''
try:
deleted_picks_from_file = self.load_deleted_picks(event_path) deleted_picks_from_file = self.load_deleted_picks(event_path)
except Exception as e:
msg = 'WARNING! Deleted picks file could not be loaded: {}' \
' Check deleted picks file in path {} before continuing'
print(msg.format(e, event_path))
self.safetyCopy(event_path)
deleted_picks_event = self.deleted_picks[event_path] deleted_picks_event = self.deleted_picks[event_path]
for pick in deleted_picks_from_file: for pick in deleted_picks_from_file:
@ -2781,11 +2788,16 @@ class MainWindow(QMainWindow):
fpath = self.get_deleted_picks_fpath(event_path) fpath = self.get_deleted_picks_fpath(event_path)
with open(fpath, 'w') as outfile: with open(fpath, 'w') as outfile:
json.dump(deleted_picks_event[event_path], outfile) json.dump(deleted_picks_event, outfile)
# clear entry for this event # clear entry for this event
self.deleted_picks[event_path] = [] self.deleted_picks[event_path] = []
def safetyCopy(self, event_path):
fpath = self.get_deleted_picks_fpath(event_path)
fpath_new = fpath.split('.json')[0] + '_copy_{}.json'.format(datetime.now()).replace(' ', '_')
shutil.move(fpath, fpath_new)
def load_deleted_picks(self, event_path): def load_deleted_picks(self, event_path):
''' Load a list with deleted picks for event in event_path from file ''' ''' Load a list with deleted picks for event in event_path from file '''
fpath = self.get_deleted_picks_fpath(event_path) fpath = self.get_deleted_picks_fpath(event_path)