update MapsResult fields (#178)

added: category facebook twitter instagram removed: links
deedy5 · Jan 28, 2024 · 53712bf · 53712bf
1 parent 5299428
commit 53712bf
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/duckduckgo_search/duckduckgo_search_async.py b/duckduckgo_search/duckduckgo_search_async.py
@@ -799,9 +799,12 @@ async def maps(
                     result.source = _normalize_url(res["url"])
                     if res["embed"]:
                         result.image = res["embed"].get("image", "")
-                        result.links = res["embed"].get("third_party_links", "")
                         result.desc = res["embed"].get("description", "")
                     result.hours = res["hours"]
+                    result.category = res["ddg_category"]
+                    result.facebook = f"www.facebook.com/profile.php?id={x}" if (x := res["facebook_id"]) else None
+                    result.instagram = f"https://www.instagram.com/{x}" if (x := res["instagram_id"]) else None
+                    result.twitter = f"https://twitter.com/{x}" if (x := res["twitter_id"]) else None
                     yield result.__dict__
                     if max_results and len(cache) >= max_results:
                         return

diff --git a/duckduckgo_search/models.py b/duckduckgo_search/models.py
@@ -16,5 +16,8 @@ class MapsResult:
     phone: Optional[str] = None
     image: Optional[str] = None
     source: Optional[str] = None
-    links: Optional[str] = None
     hours: Optional[Dict[str, str]] = None
+    category: Optional[str] = None
+    facebook: Optional[str] = None
+    instagram: Optional[str] = None
+    twitter: Optional[str] = None