init commit

mayo.py

@@ -0,0 +1,111 @@
+import requests
+from bs4 import BeautifulSoup
+import re
+
+class MayoSession:
+    def __init__(self, base_url, login, password, db="1"):
+        """
+        base_url: np. 'http://192.168.0.152/mayo2'
+        login, password: dane logowania
+        db: numer bazy (np. "1" = Mayones 2)
+        """
+        self.session = requests.Session()
+        self.base_url = base_url
+        self.login_url = f"{self.base_url}/login.php"
+        self.credentials = {
+            "login": login,
+            "pass": password,
+            "baza": db
+        }
+
+    def login(self):
+        """Loguje się do systemu lokalnego."""
+        r = self.session.post(self.login_url, data=self.credentials)
+        if "Zaloguj się" in r.text or "login" in r.url:
+            raise Exception("Nie udało się zalogować do Mayo.")
+        print("✅ Zalogowano poprawnie do systemu Mayo.")
+
+    # def get_order_info(self, url):
+    #     """
+    #     Pobiera dane z podanej strony zamówienia:
+    #     - numer zamówienia
+    #     - model gitary
+    #     """
+    #     r = self.session.get(url)
+    #     r.encoding = "utf-8"
+    #     soup = BeautifulSoup(r.text, "html.parser")
+
+    #     # nr zamówienia
+    #     order_span = soup.find("span", class_="czarnobiale")
+    #     order_number = order_span.get_text(strip=True) if order_span else None
+
+    #     # model gitary
+    #     input_tag = soup.find("input", {"name": "s_nr_kat"})
+    #     model = input_tag.get("value").strip() if input_tag else None
+
+    #     return {"order_number": order_number, "model": model}
+    
+    def get_order_info(self, url):
+        """
+        Pobiera dane z podanej strony zamówienia:
+        - numer zamówienia
+        - model
+        - kolory (Top, Korpus, Szyjka, Główka)
+        - wykończenie
+        """
+        r = self.session.get(url)
+        r.encoding = "utf-8"
+        soup = BeautifulSoup(r.text, "html.parser")
+
+        # --- nr zamówienia ---
+        order_span = soup.find("span", class_="czarnobiale")
+        order_number = order_span.get_text(strip=True) if order_span else None
+
+        # --- model gitary ---
+        input_tag = soup.find("input", {"name": "s_nr_kat"})
+        model = input_tag.get("value").strip() if input_tag else None
+
+        # --- kolory i wykończenie ---
+        color_sections = {}
+        # Znajdź wszystkie <p> zawierające <a> z tekstem "KOLOR"
+        for p in soup.find_all("p"):
+            a_tag = p.find("a")
+            if not a_tag:
+                continue
+            title = a_tag.get_text(strip=True)
+            if title.startswith("KOLOR"):
+                # np. "KOLOR - Top"
+                label = title.replace("KOLOR - ", "").strip()
+
+                # znajdź <td> z wartością koloru (niedaleko tego <p>)
+                td = p.find_parent("td")
+                if td:
+                    # przejdź do następnego <td>, tam jest <span> z kolorem
+                    next_td = td.find_next_sibling("td")
+                    if next_td:
+                        span = next_td.find("span")
+                        if span:
+                            text = span.get_text(" ", strip=True)
+                            # usuń ewentualne znaki nadmiarowe
+                            text = re.sub(r"\s+", " ", text)
+                            color_match = re.search(r"^\s*-\s*([A-Z0-9-]+)", text)
+                            if color_match:
+                                text = color_match.group(1)
+                            color_sections[label] = text
+
+        # --- normalizacja nazewnictwa ---
+        color_top = color_sections.get("Top")
+        color_body = color_sections.get("Korpus")
+        color_neck = color_sections.get("Szyjka")
+        color_head = color_sections.get("Główka")
+        finish = color_sections.get("Wykończenie [K/C]")
+
+        return {
+            "order_number": order_number,
+            "model": model,
+            "color_top": color_top,
+            "color_body": color_body,
+            "color_neck": color_neck,
+            "color_head": color_head,
+            "finish": finish,
+        }