From 02b00c50ca7248392357ee2bdac692802968cdc6 Mon Sep 17 00:00:00 2001 From: crab85193 Date: Mon, 18 Dec 2023 21:15:03 +0900 Subject: [PATCH] fix --- src/google_api_controller.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/src/google_api_controller.py b/src/google_api_controller.py index 079d368..56c0226 100644 --- a/src/google_api_controller.py +++ b/src/google_api_controller.py @@ -21,7 +21,7 @@ def download_place_photo(self, photo_reference, file_name=None): if response.status_code == 200: if file_name: - f_name = os.environ.get("SAVE_IMG_DIR") + "/" + file_name + f_name = os.environ.get("SAVE_IMG_DIR") + "/" + file_name + ".jpg" else: f_name = os.environ.get("SAVE_IMG_DIR") + "/" + photo_reference + ".jpg" @@ -31,9 +31,9 @@ def download_place_photo(self, photo_reference, file_name=None): else: print("Error: Unable to download the image.") - def get_store_photo_url(self, photo_reference): - self.download_place_photo(photo_reference) - return f"{os.environ.get('IMAGE_SERVER_PROTOCOL')}://{os.environ.get('IMAGE_SERVER_ADDRESS')}/static/{photo_reference}.jpg" + def get_store_photo_url(self, photo_reference, place_id): + self.download_place_photo(photo_reference, place_id) + return f"{os.environ.get('IMAGE_SERVER_PROTOCOL')}://{os.environ.get('IMAGE_SERVER_ADDRESS')}/static/{place_id}.jpg" def search_store(self, keyword): location = geocoder.ip('me').latlng @@ -57,7 +57,7 @@ def search_store(self, keyword): except Exception as e: store_info["open"] = None try: - store_info["photos"] = self.get_store_photo_url(place["photos"][0]["photo_reference"]) + store_info["photos"] = self.get_store_photo_url(place["photos"][0]["photo_reference"], place["place_id"]) except Exception as e: print(e) store_info["photos"] = ""