parameterized number of days to be pulled

lmEshoo · Jan 14, 2021 · 7f17062 · 7f17062
1 parent d508300
commit 7f17062
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/Makefile b/Makefile
@@ -1,6 +1,7 @@
 USERNAME=lmestar
 CONTAINER_NAME=wz
 IMAGE=$(USERNAME)/$(CONTAINER_NAME):0.2
+DAYS?=8
 
 all: run
 
@@ -17,6 +18,7 @@ up:
 	-p 5901:5901 \
 	-v /dev/shm:/dev/shm \
 	-e USER=${USER} \
+	-e DAYS=${DAYS} \
 	-e TZ=America/Los_Angeles \
 	--name $(CONTAINER_NAME) $(IMAGE) \
 	bash -c "sleep 2 && python3 wz.py "

diff --git a/README.md b/README.md
@@ -21,7 +21,7 @@ docker pull lmestar/wz:0.2
 
 ### Run 
 ```
-make USER=<Enter your PSN username>
+make USER=<Enter your PSN username> DAYS=<# of past days to collect>
 ```
 
 ## Visualize

diff --git a/wz.py b/wz.py
@@ -36,7 +36,7 @@ def get_basics(self):
                 sleep (1)
                 pass
 
-        for day in range(1,12):
+        for day in range(1,int(os.getenv('DAYS'))+1):
             print(day)
             cod_tracker_date                = self.driver.find_element_by_xpath('//*[@id="app"]/div[2]/div[2]/div/main/div[2]/div[3]/div[2]/div/div[2]/div['+str(day)+']/div[1]/h3').text.split('\n')[0]
             cod_tracker_games_per_day       = self.driver.find_element_by_xpath('//*[@id="app"]/div[2]/div[2]/div/main/div[2]/div[3]/div[2]/div/div[2]/div['+str(day)+']/div[1]/h3').text.split('\n')[1]
@@ -62,7 +62,7 @@ def get_basics(self):
                 cod_tracker_match_link_list.append(cod_tracker_match_link)
 
         self.driver.quit()
-        
+
         return cod_tracker_games_per_day_list, cod_tracker_date_list, cod_tracker_mode_list, cod_tracker_match_placement_list, cod_tracker_time_list, cod_tracker_kills_list, cod_tracker_damage_list, cod_tracker_match_link_list
 
 class SbmmWz():
-Original file line number
+Diff line change
@@ Expand Up / @@ -21,7 +21,7 @@ docker pull lmestar/wz:0.2 @@
     ### Run
     ```
-    make USER=<Enter your PSN username>
+    make USER=<Enter your PSN username> DAYS=<# of past days to collect>
     ```
     ## Visualize
@@ Expand Down @@