Changed daily field to number of days

Author: aheldes

component_config/configSchema.json

@@ -4,35 +4,41 @@
   "required": [
     "#client_id",
     "#client_secret",
-    "daily_load",
+    "days",
     "manual_authentication",
     "endpoint"
   ],
   "properties": {
     "#client_id": {
       "type": "string",
       "format": "password",
-      "title": "Allegro Client ID"
+      "title": "Allegro Client ID",
+      "propertyOrder": 1
     },
     "#client_secret": {
       "type": "string",
       "format": "password",
-      "title": "Allegro Client Secret"
+      "title": "Allegro Client Secret",
+      "propertyOrder": 2
     },
     "endpoint": {
       "title": "Endpoint",
       "type": "string",
       "enum": [
         "Billing entries"
-      ]
+      ],
+      "propertyOrder": 3
     },
-    "daily_load": {
-      "type": "boolean",
-      "title": "Daily load"
+    "days": {
+      "type": "integer",
+      "title": "Days",
+      "description": "How many days to load",
+      "propertyOrder": 4
     },
     "manual_authentication": {
       "type": "boolean",
-      "title": "Manual Authentication"
+      "title": "Manual Authentication",
+      "propertyOrder": 5
     }
   }
 }

src/component.py

@@ -17,7 +17,7 @@
 KEY_CLIENT_ID = '#client_id'
 KEY_CLIENT_SECRET = '#client_secret'
 ENDPOINTS = 'endpoint'
-DAILY = 'daily_load'
+DAYS = 'days'
 AUTHENTICATION = 'manual_authentication'
 
 CODE_URL = "https://allegro.pl/auth/oauth/device"
@@ -59,7 +59,7 @@ def run(self):
         self.client_ID = params.get(KEY_CLIENT_ID)
         self.client_secret = params.get(KEY_CLIENT_SECRET)
         self.endpoint = params.get(ENDPOINTS)
-        self.daily = params.get(DAILY)
+        self.days = params.get(DAYS)
         self.authentication = params.get(AUTHENTICATION)
 
         # Get state file
@@ -223,13 +223,11 @@ def parse_biling_entries(data):
             return df
 
         logging.info('Getting dates.')
-        if self.daily:
-            # Set start and stop date to yesterday
-            start_date = stop_date = datetime.today().date() - timedelta(days=1)
-        else:
-            # Set start to 1.1.2020 and stop to yesterday
-            start_date = date(year=2020, month=1, day=1)
-            stop_date = datetime.today().date() - timedelta(days=1)
+
+        # Set start to 1.1.2020 and stop to yesterday
+        today = datetime.today().date()
+        start_date = today  - timedelta(days=1)
+        stop_date = datetime.today().date() - timedelta(days=self.days)
 
         # Get date list
         date_list = date_range_list(start_date, stop_date)