OwlBoard/db-manager
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Releases Activity

Wrap _addToDb functionality in a try:except block

Browse Source
This commit is contained in:
Fred Boniface 2023-06-04 18:39:06 +01:00
parent e17d03b376
commit 89574d4bbd
1 changed files with 17 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/timetable.py
View File

@ -141,6 +141,7 @@ def insertSchedule(sch_record):
return document return document
def _insertToDb(data :list, type :str): def _insertToDb(data :list, type :str):
try:
if type == "full": if type == "full":
mongo.dropCollection("timetable") mongo.dropCollection("timetable")
mongo.putTimetable(data) mongo.putTimetable(data)
@ -151,8 +152,12 @@ def _insertToDb(data :list, type :str):
singleList = [item] singleList = [item]
mongo.putTimetable(singleList) mongo.putTimetable(singleList)
elif item['transactionType'] == "Delete": elif item['transactionType'] == "Delete":
mongo.deleteTimetableData({'trainUid': item['trainUid']}) mongo.deleteTimetableData({'trainUid': item['trainUid']}) ## Also need to consider the STP indicator and end date here else I am deleting LTP services when an STP is meant to be deleted.
return True #If Successful else False return True #If Successfuls
except Exception as e:
log.out("timetable._insertToDb: Error inserting timetable data", "ERR")
log.out(f"timteable._insertToDb: {e}")
return False # If error inserting timetable
def _helpParseStops(schedule_segment): def _helpParseStops(schedule_segment):
stops = [] stops = []