Use db data to start app

Author: Jack Li

storage/base.py

@@ -1,4 +1,6 @@
 from pymongo import collection
+from uuid import uuid4
+from datetime import datetime
 
 def insert_one__(coll: collection.Collection, doc: dict):
    return coll.insert_one(doc)
@@ -23,3 +25,9 @@ def find_one__(coll: collection.Collection, filter: dict, projection):
 
 def find_many__(coll: collection.Collection, filter: dict, projection, **kwargs):
    return coll.find(filter=filter, projection=projection, **kwargs)
+
+def gen_additional_attrs():
+   return {
+       "_id": str(uuid4()),
+       "last_modified": datetime.today().replace(microsecond=0)
+   }

storage/storage.py

@@ -1,20 +1,21 @@
 from utils import get_db_handle
 from .base import *
-from uuid import uuid4
 
 # insert one
 def insert_one(collection_name, doc: dict, db_name="tickets"):
    db = get_db_handle(db_name)
-   doc["_id"] = str(uuid4())
    coll = db[collection_name]
+   # additional attributes
+   doc.update(gen_additional_attrs())
    return insert_one__(coll, doc)
 
 # insert many
 def insert_many(collection_name, docs: list, db_name="tickets"):
    db = get_db_handle(db_name)
-   for doc in docs:
-      doc["_id"] = str(uuid4())
    coll = db[collection_name]
+   # additional attributes
+   for doc in docs:
+      doc.update(gen_additional_attrs())
    return insert_many__(coll, docs)
 
 # delete one

ticketscraping/scraping.py

@@ -28,6 +28,7 @@ def renew_reese84_token(self):
         """
         This method should not be called directly.
         """
+        print("renewing token")
         self.token_semaphore.acquire() # invalidate a token
         self.reese84_token = getReese84Token()
         self.token_semaphore.release()
@@ -44,9 +45,10 @@ def start(self):
 
 
 class TicketScraping():
-    def __init__(self, token_generator: Reese84TokenUpdating, event_id = constants.EVENT_ID, num_seats=2, price_range=(0, 200)):
+    def __init__(self, token_generator: Reese84TokenUpdating, event_id, subscribe_id, num_seats=2, price_range=(0, 200)):
         self.is_running = False
         self.event_id = event_id
+        self.subscribe_id = subscribe_id
         self.num_seats = num_seats
         self.price_range = price_range
         self.token_gen = token_generator
@@ -90,10 +92,10 @@ def start():
 
     # ticket scraping threads
     scraping_list = []
-    events = find_many(constants.DATABASE["EVENTS"], {}, {"_id": False})
+    events = find_many(constants.DATABASE["EVENTS"], {})
 
     for evt in events:
-        ticket_scraping = TicketScraping(reese_token_gen, evt["tm_event_id"], evt["ticket_num"], evt["price_range"])
+        ticket_scraping = TicketScraping(reese_token_gen, evt["tm_event_id"], evt["_id"], evt["ticket_num"], evt["price_range"])
         print(ticket_scraping.initialDelay, "s")
         serverThread_ticket_scraping = Thread(target=ticket_scraping.start)
         scraping_list.append(serverThread_ticket_scraping)

tmtracker/settings.py

@@ -27,28 +27,34 @@
 
 ALLOWED_HOSTS = []
 
-
 # Application definition
 
 INSTALLED_APPS = [
-    'django.contrib.admin',
-    'django.contrib.auth',
+    # 'django.contrib.admin',
+    # 'django.contrib.auth',
     'django.contrib.contenttypes',
-    'django.contrib.messages',
+    # 'django.contrib.messages',
     'django.contrib.staticfiles',
-    'startupApp'
+    'startupApp',
+    'trackerapi',
 ]
 
 MIDDLEWARE = [
     'django.middleware.security.SecurityMiddleware',
-    'django.contrib.sessions.middleware.SessionMiddleware',
+    # 'django.contrib.sessions.middleware.SessionMiddleware',
     'django.middleware.common.CommonMiddleware',
-    'django.middleware.csrf.CsrfViewMiddleware',
-    'django.contrib.auth.middleware.AuthenticationMiddleware',
-    'django.contrib.messages.middleware.MessageMiddleware',
+    # 'django.middleware.csrf.CsrfViewMiddleware',
+    # 'django.contrib.auth.middleware.AuthenticationMiddleware',
+    # 'django.contrib.messages.middleware.MessageMiddleware',
     'django.middleware.clickjacking.XFrameOptionsMiddleware',
 ]
 
+REST_FRAMEWORK = {
+    'DEFAULT_AUTHENTICATION_CLASSES': [],
+    'DEFAULT_PERMISSION_CLASSES': [],
+    'UNAUTHENTICATED_USER': None
+}
+
 ROOT_URLCONF = 'tmtracker.urls'
 
 TEMPLATES = [

tmtracker/urls.py

@@ -14,8 +14,8 @@
     2. Add a URL to urlpatterns:  path('blog/', include('blog.urls'))
 """
 from django.contrib import admin
-from django.urls import path
+from django.urls import path, include
 
 urlpatterns = [
-    path('admin/', admin.site.urls),
+    path('tracker-api/', include('trackerapi.urls'))
 ]

trackerapi/__init__.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

trackerapi/admin.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class TrackerapiConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'trackerapi'

trackerapi/apps.py

@@ -0,0 +1,3 @@
+from django.db import models
+
+# Create your models here.