Browse Source

Add resource_id to track

JoostSijm 5 years ago
parent
commit
bdcbf41268
2 changed files with 3 additions and 2 deletions
  1. 1 1
      hvs/__main__.py
  2. 2 1
      hvs/app.py

+ 1 - 1
hvs/__main__.py

@@ -11,7 +11,7 @@ from hvs.app import download_resources, need_refill, refill, save_resources, pri
 def job_check_resources(state_id, capital_id, resource_id):
     """Check resources and refill if necessary"""
     regions = download_resources(state_id, resource_id)
-    save_resources(state_id, regions)
+    save_resources(state_id, regions, resource_id)
     print_resources(regions)
     if need_refill(regions):
         random_time = timedelta(seconds=random.randint(0, 300))

+ 2 - 1
hvs/app.py

@@ -128,11 +128,12 @@ def refill(state_id, capital_id, resource_id):
         print(result.text)
 
 
-def save_resources(state_id, regions):
+def save_resources(state_id, regions, resource_id):
     """Save resources to database"""
     session = Session()
     resource_track = ResourceTrack()
     resource_track.state_id = state_id
+    resource_track.resource_id = resource_id
     session.add(resource_track)
     session.commit()