From 728b954643af33eee4842d551fba2d6d78a755f2 Mon Sep 17 00:00:00 2001
From: Helldragon67 <kilian.scheidecker@orange.fr>
Date: Wed, 28 Aug 2024 18:08:09 +0200
Subject: [PATCH] added image and website urls
---
backend/src/structs/landmark.py | 3 +++
backend/src/utils/landmarks_manager.py | 25 ++++++++++++++++++++++---
2 files changed, 25 insertions(+), 3 deletions(-)
diff --git a/backend/src/structs/landmark.py b/backend/src/structs/landmark.py
index 2f22d42..7e0cc97 100644
--- a/backend/src/structs/landmark.py
+++ b/backend/src/structs/landmark.py
@@ -15,8 +15,11 @@ class Landmark(BaseModel) :
attractiveness : int
n_tags : int
image_url : Optional[str] = None # TODO future
+ website_url : Optional[str] = None
+ wikipedia_url : Optional[str] = None
description : Optional[str] = None # TODO future
duration : Optional[int] = 0 # TODO future
+ name_en : Optional[str] = None
# Unique ID of a given landmark
uuid: str = Field(default_factory=uuid4) # TODO implement this ASAP
diff --git a/backend/src/utils/landmarks_manager.py b/backend/src/utils/landmarks_manager.py
index 5f5e82e..db300f7 100644
--- a/backend/src/utils/landmarks_manager.py
+++ b/backend/src/utils/landmarks_manager.py
@@ -290,7 +290,10 @@ class LandmarkManager:
elem_type = landmarktype # Add the landmark type as 'sightseeing,
n_tags = len(elem.tags().keys()) # Add number of tags
score = n_tags**self.tag_exponent # Add score
-
+ website_url = None
+ wikpedia_url = None
+ image_url = None
+ name_en = None
# remove specific tags
skip = False
@@ -304,7 +307,7 @@ class LandmarkManager:
if "wiki" in tag:
score += self.wikipedia_bonus # wikipedia entries count more
-
+
# if tag == "wikidata":
# Q = elem.tag('wikidata')
# site = Site("wikidata", "wikidata")
@@ -331,6 +334,18 @@ class LandmarkManager:
if tag == "building" and elem.tag('building') in ['retail', 'supermarket', 'parking']:
skip = True
break
+
+
+ # Get additional information
+ # if tag == 'wikipedia' :
+ # wikpedia_url = elem.tag('wikipedia')
+ if tag in ['website', 'contact:website'] :
+ website_url = elem.tag(tag)
+ if tag == 'image' :
+ image_url = elem.tag('image')
+ if tag =='name:en' :
+ name_en = elem.tag('name:en')
+
if skip:
continue
@@ -356,7 +371,11 @@ class LandmarkManager:
attractiveness=score,
must_do=False,
n_tags=int(n_tags),
- duration = duration
+ duration = duration,
+ name_en=name_en,
+ image_url=image_url,
+ # wikipedia_url=wikpedia_url,
+ website_url=website_url
)
return_list.append(landmark)