Skip to content

Commit 8cad815

Browse files
update storage.py and scraping.py
1 parent 1875b2c commit 8cad815

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

storage/storage.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,19 @@
11
from ..utils import get_db_handle
22
from .base import *
3+
from uuid import uuid4
34

45
# insert one
56
def insert_one(collection_name, doc: dict, db_name="tickets"):
67
db = get_db_handle(db_name)
8+
doc["_id"] = str(uuid4())
79
coll = db[collection_name]
810
return __insert_one__(coll, doc)
911

1012
# insert many
1113
def insert_many(collection_name, docs: list, db_name="tickets"):
1214
db = get_db_handle(db_name)
15+
for doc in docs:
16+
doc["_id"] = str(uuid4())
1317
coll = db[collection_name]
1418
return __insert_many__(coll, docs)
1519

@@ -31,6 +35,7 @@ def find_one_and_replace(collection_name, filter: dict, new_doc: dict, db_name="
3135
coll = db[collection_name]
3236
return __find_one_and_replace__(coll, filter, new_doc)
3337

38+
# find one and delete
3439
def find_one_and_delete(collection_name, filter: dict, db_name="tickets"):
3540
db = get_db_handle(db_name)
3641
coll = db[collection_name]
@@ -46,4 +51,4 @@ def find_one(collection_name, filter: dict, projection=None, db_name="tickets"):
4651
def find_many(collection_name, filter: dict, projection=None, db_name="tickets", **kwargs):
4752
db = get_db_handle(db_name)
4853
coll = db[collection_name]
49-
return __find_many__(coll, filter, projection, **kwargs)
54+
return __find_many__(coll, filter, projection, **kwargs)

ticketscraping/scraping.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ def start(self):
7777
self.is_running = True
7878
self.ticket_scraping()
7979
# randomize start time to scatter out event of API fetching
80-
time.sleep(self.initialDelay)
80+
time.sleep(self.initialDelay)
8181
self.scheduler.run()
8282

8383

0 commit comments

Comments
 (0)