Merge pull request #9 from OmarMAshour/labels

api barcode first commit

data_fetcher/fetcher.py

@@ -17,14 +17,34 @@ def __init__(self) -> None:
 
         api_details = dict(self.api_conf.items("Wine Info"))
 
-        self.url = api_details['url']
+        self.info_url = api_details['info_url']
+        self.barcode_url = api_details["barcode_url"]
         self.api_key = api_details['api_key']
 
         # read keys configuration for label and info sheet
         self.keys_conf = configparser.RawConfigParser()
         self.keys_conf.read("../conf/keys.conf")
         self.label_keys = ast.literal_eval(self.keys_conf.get("Keys", "label"))
 
+    def find_article_id_by_barcode(self, barcode: str) -> str:
+        return self.find_by_barcode(barcode=barcode)["code"]
+
+    def find_by_barcode(self, barcode: str) -> Dict:
+        """
+        Fetch wine brief by a barcode
+        """
+
+        params = {"lang": "de", "gtin": barcode}
+        headers = {"APIKey": self.api_key}
+
+        response = requests.get(
+            self.barcode_url,
+            params=params,
+            headers=headers
+        )
+
+        return response.json()
+
     def find_for_labels(self, article_ids: List[str]) -> List[Dict]:
         return self.find_by_article_ids(article_ids, self.label_keys)
 
@@ -37,7 +57,7 @@ def find_by_article_ids(self, article_ids: List[str], keys: List[str] = None) ->
         headers = {"APIKey": self.api_key}
 
         response = requests.get(
-            self.url,
+            self.info_url,
             params=params,
             headers=headers
         )
@@ -52,5 +72,4 @@ def find_by_article_ids(self, article_ids: List[str], keys: List[str] = None) ->
 
 if __name__ == "__main__":
     fetcher = Fetcher()
-    items = fetcher.find_for_labels(article_ids=["1007906010"])
-    print(items)
+    print(fetcher.find_article_id_by_barcode("3760126362587"))