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)