des expérimentations pour avoir un meilleur import des événements facebook

experimentations/get_facebook_event.py

@@ -8,37 +8,100 @@ from selenium import webdriver
 from selenium.webdriver.chrome.service import Service
 from selenium.webdriver.chrome.options import Options
 
+
 from bs4 import BeautifulSoup
 
 import json
 
+class SimpleEvent:
+
+    def __init__(self, data):
+        self.elements = {}
+
+        for key in ["id", "start_timestamp", "end_timestamp"]:
+            self.elements[key] = data[key] if key in data else None
+
+        if "parent_event" in data:
+            self.parent = SimpleEvent(data["parent_event"])
+
+
 class Event:
 
     name = "event"
-    keys = ["start_time_formatted", 'start_timestamp', 'is_past', "name", "price_info", "cover_media_renderer", "event_creator", "id", "day_time_sentence", "event_place", "comet_neighboring_siblings"]
+    keys = [
+            ["start_time_formatted", 'start_timestamp', 
+             'is_past', 
+             "name", 
+             "price_info", 
+             "cover_media_renderer", 
+             "event_creator", 
+             "id", 
+             "day_time_sentence", 
+             "event_place", 
+             "comet_neighboring_siblings"],
+            ["event_description"]
+    ]
+    rules = {
+        "event_description": { "description": ["text"]},
+        "cover_media_renderer": {"image_alt": ["cover_photo", "photo", "accessibility_caption"], "image": ["cover_photo", "photo", "full_image", "uri"]},
+        "event_creator": { "event_creator_name": ["name"], "event_creator_url": ["url"] },
+        "event_place": {"event_place_name": ["name"] }
+    }
 
-    def __init__(self, event):
-        self.data = event
+    def __init__(self, i, event):
+        self.fragments = {}
+        self.elements = {}
+        self.neighbor_events = None
+        self.add_fragment(i, event)
+
+    def add_fragment(self, i, event):
+        self.fragments[i] = event
+
+        for k in Event.keys[i]:
+            if k == "comet_neighboring_siblings":
+                self.get_neighbor_events(event[k])
+            elif k in Event.rules:
+                for nk, rule in Event.rules[k].items():
+                    c = event[k]
+                    for ki in rule:
+                        c = c[ki]
+                    self.elements[nk] = c
+            else:
+                self.elements[k] = event[k]
+
+
+    def get_neighbor_events(self, data):
+        self.neighbor_events = [SimpleEvent(d) for d in data]
 
     def __str__(self):
-        return self.data["name"]
+        return str(self.elements) + "\n Neighbors: " + ", ".join([ne.elements["id"] for ne in self.neighbor_events])
 
-    def find_event_in_array(array):
+    def consolidate_current_event(self):
+        if self.neighbor_events is not None and "id" in self.elements:
+            id = self.elements["id"]
+            for ne in self.neighbor_events:
+                if ne.elements["id"] == id:
+                    self.elements["end_timestamp"] = ne.elements["end_timestamp"]
+
+    def find_event_fragment_in_array(array, event):
         if isinstance(array, dict):
-            #print([k for k in array])
-            if len(Event.keys) == len([k for k in Event.keys if k in array]):
-                return Event(array)
-            else:
-                for k in array:
-                    v = Event.find_event_in_array(array[k])
-                    if v != None:
-                        return v
+
+            for i, ks in enumerate(Event.keys):
+                if len(ks) == len([k for k in ks if k in array]):
+                    if event is None:
+                            event = Event(i, array,)
+                    else:
+                        event.add_fragment(i, array)
+                else:
+                    for k in array:
+                        event = Event.find_event_fragment_in_array(array[k], event)
         elif isinstance(array, list):
             for e in array:
-                    v = Event.find_event_in_array(e)
-                    if v != None:
-                        return v
-        return None
+                event = Event.find_event_fragment_in_array(e, event)
+
+        if event is not None:
+            event.consolidate_current_event()
+        return event
 
 
 #url="https://www.facebook.com/events/ical/export/?eid=2294200007432315"
@@ -53,7 +116,7 @@ hash = result.hexdigest()
 filename = os.path.join(cachedir, hash + ".html")
 
 if os.path.isfile(filename):
-    #print("Use cache")
+    # print("Use cache")
     with open(filename) as f:
         doc = "\n".join(f.readlines())
 else:
@@ -77,9 +140,12 @@ else:
 
 soup = BeautifulSoup(doc)
 
+event = None
 for json_script in soup.find_all('script', type="application/json"):
     json_txt = json_script.get_text()
     json_struct = json.loads(json_txt)
-    event = Event.find_event_in_array(json_struct)
-    if event != None:
-        print(event)
+
+    event = Event.find_event_fragment_in_array(json_struct, event)
+
+print(event)
+