From 9bf5155e9f11a49d127e7c41c74720e97ca3102f Mon Sep 17 00:00:00 2001
From: Elmar Kresse <elmar.kresse@stud.htwk-leipzig.de>
Date: Sun, 16 Feb 2025 22:40:35 +0100
Subject: [PATCH] feat: add VLW owner handling and image scraping logic in LWB
 and VLW scrapers

---
 src/lwb/scrape_image.py | 7 +++++++
 src/vlw/scraper.py      | 1 +
 2 files changed, 8 insertions(+)

diff --git a/src/lwb/scrape_image.py b/src/lwb/scrape_image.py
index 6a0147d..60fccff 100644
--- a/src/lwb/scrape_image.py
+++ b/src/lwb/scrape_image.py
@@ -11,6 +11,13 @@ def scrape_image(url, owner):
             print(f"Fehler beim Abrufen von BGL: {response.status_code}")
             # return empty image
             return b''
+        
+    if owner == "VLW":
+        response = session.get(url)
+        if response.status_code != 200:
+            print(f"Fehler beim Abrufen von VLW: {response.status_code}")
+            # return empty image
+            return b''
 
     if owner == "LWB":  
         response = session.get(url, headers=EASYSQUARE_HEADERS, params=EASYSQUARE_PARAMS)
diff --git a/src/vlw/scraper.py b/src/vlw/scraper.py
index 923ad80..ec9a8ba 100644
--- a/src/vlw/scraper.py
+++ b/src/vlw/scraper.py
@@ -72,6 +72,7 @@ def scrape_vlw():
             "warm_rent": warm_rent,
             "availability": availability,
             "image_url": image_url,
+            "owner": "VLW",
         })
 
     return properties
\ No newline at end of file