Changeset 47f4eaf for phonelux_scrappers/scrappers/a1_scrapper.py

Timestamp:

11/20/22 16:34:52 (2 years ago)

Author:

Marko <Marko@…>

Branches:

master

Parents:

ffd50db

Message:

Final features implemented

File:

• phonelux_scrappers/scrappers/a1_scrapper.py (modified) (2 diffs)

phonelux_scrappers/scrappers/a1_scrapper.py

@@ +1 @@
+import traceback
 import unicodedata
 from datetime import datetime
@@ -18 +19 @@
 is_validated = False
 
-# A1 phone offers that are already in database
+# Call to read the configuration file and connect to database
+cinfo = config_read.get_databaseconfig("../postgresdb.config")
+db_connection = psycopg2.connect(
+    database=cinfo[0],
+    host=cinfo[1],
+    user=cinfo[2],
+    password=cinfo[3]
+)
+cur = db_connection.cursor()
 
-offers = json.loads(unicodedata.normalize('NFKD', requests.get('http://localhost:8080/phoneoffer/shop/a1').text))
+try:
+    # A1 phone offers that are already in database
+    offers = json.loads(unicodedata.normalize('NFKD', requests.get('http://localhost:8080/phoneoffer/shop/a1').text))
 
-database_offers = []
+    database_offers = []
 
-for offer in offers:
-    phoneOffer = PhoneOffer(offer['id'], offer['offer_shop'], offer['offer_name'], offer['price'],
-                            offer['ram_memory'],
-                            offer['rom_memory'], offer['color'], offer['front_camera'], offer['back_camera'],
-                            offer['chipset'], offer['battery'], offer['operating_system'], offer['cpu'],
-                            offer['image_url'],
-                            offer['offer_url'], offer['last_updated'], offer['is_validated'],
-                            offer['offer_description'],
-                            offer['offer_shop_code'])
-    database_offers.append(phoneOffer)
+    for offer in offers:
+        phoneOffer = PhoneOffer(offer['id'], offer['offer_shop'], offer['offer_name'], offer['price'],
+                                offer['ram_memory'],
+                                offer['rom_memory'], offer['color'], offer['front_camera'], offer['back_camera'],
+                                offer['chipset'], offer['battery'], offer['operating_system'], offer['cpu'],
+                                offer['image_url'],
+                                offer['offer_url'], offer['last_updated'], offer['is_validated'],
+                                offer['offer_description'],
+                                offer['offer_shop_code'])
+        database_offers.append(phoneOffer)
 
-a1_url = 'https://www.a1.mk/webshop/mk/phones'
+    a1_url = 'https://www.a1.mk/webshop/mk/phones'
 
-response1 = requests.get(a1_url)
-soup1 = BeautifulSoup(response1.content, 'html.parser')
+    response1 = requests.get(a1_url)
+    soup1 = BeautifulSoup(response1.content, 'html.parser')
 
-phones = soup1.find('main', {'class', 'gsm-advisor-grid phones'}).find('div', {'class', 'd-flex'}) \
-    .find_all('div', {'class', 'dvc-idtfr by4'})
+    phones = soup1.find('main', {'class', 'gsm-advisor-grid phones'}).find('div', {'class', 'd-flex'}) \
+        .find_all('div', {'class', 'dvc-idtfr by4'})
 
-new_offers = []
+    new_offers = []
 
-for phone in phones:
-    brand = phone.get('data-brand').strip()
-    offer_name = brand + " " + phone.get('data-model').strip()
+    for phone in phones:
+        brand = phone.get('data-brand').strip()
+        offer_name = brand + " " + phone.get('data-model').strip()
 
-    # if brand not in offer_name:
-    #     offer_name = brand+" "+offer_name
+        # if brand not in offer_name:
+        #     offer_name = brand+" "+offer_name
 
-    offer_shop_code = phone.get('data-productid').strip()
-    offer_url = phone.find('a', {'class', 'device-link'}).get('href')
-    image_url = phone.get('data-image')
+        offer_shop_code = phone.get('data-productid').strip()
+        offer_url = phone.find('a', {'class', 'device-link'}).get('href')
+        image_url = phone.get('data-image')
 
-    response2 = requests.get(offer_url)
-    soup2 = BeautifulSoup(response2.content, 'html.parser')
+        response2 = requests.get(offer_url)
+        soup2 = BeautifulSoup(response2.content, 'html.parser')
 
-    temp_prices = soup2.find('div', {'class': 'ured-tabs-content'}) \
-        .find('div', {'class': 'cenovnik-secondary d-flex justify-content-between'}).find_all('div')
+        temp_prices = soup2.find('div', {'class': 'ured-tabs-content'}) \
+            .find('div', {'class': 'cenovnik-secondary d-flex justify-content-between'}).find_all('div')
 
-    # offer price
-    price = None
-    for temp_price in temp_prices:
-        if 'Цена само за уред' in temp_price.get_text().strip():
-            price = int(temp_price.get_text().replace('Цена само за уред', '')
-                        .replace('Одбери', '').replace('денари', '').replace('.', '').strip())
+        # offer price
+        price = None
+        for temp_price in temp_prices:
+            if 'Цена само за уред' in temp_price.get_text().strip():
+                price = int(temp_price.get_text().replace('Цена само за уред', '')
+                            .replace('Одбери', '').replace('денари', '').replace('.', '').strip())
 
-    colors_section = soup2.find('div', {'id': 'hero'}).find('div', {'class': 'widget'}).find_all('label')
+        colors_section = soup2.find('div', {'id': 'hero'}).find('div', {'class': 'widget'}).find_all('label')
 
-    temp_colors = []
-    for color_section in colors_section:
-        temp_colors.append(color_section.get('data-content'))
+        temp_colors = []
+        for color_section in colors_section:
+            temp_colors.append(color_section.get('data-content'))
 
-    color = ','.join(temp_colors)  # colors available for the offer
+        color = ','.join(temp_colors)  # colors available for the offer
 
-    phone_description = soup2.find('div', {'class': 'desc section'}).find('p').get_text().strip()
+        phone_description = soup2.find('div', {'class': 'desc section'}).find('p').get_text().strip()
 
-    table_rows = soup2.find('table', {'class': 'table karakteristiki'}).find_all('tr')
+        table_rows = soup2.find('table', {'class': 'table karakteristiki'}).find_all('tr')
 
-    back_camera = None
-    operating_system = None
-    cpu = None
-    rom_memory = None
-    ram_memory = None
-    battery = None
-    front_camera = None
-    chipset = None
-    offer_description = None
+        back_camera = None
+        operating_system = None
+        cpu = None
+        rom_memory = None
+        ram_memory = None
+        battery = None
+        front_camera = None
+        chipset = None
+        offer_description = None
 
-    for row in table_rows:
-        if 'Камера' in row.get_text().strip():
-            back_camera = row.get_text().replace('Камера', '').strip()
+        for row in table_rows:
+            if 'Камера' in row.get_text().strip():
+                back_camera = row.get_text().replace('Камера', '').strip()
 
-        if 'Оперативен систем' in row.get_text().strip():
-            operating_system = row.get_text().replace('Оперативен систем', '').strip()
+            if 'Оперативен систем' in row.get_text().strip():
+                operating_system = row.get_text().replace('Оперативен систем', '').strip()
 
-        if 'CPU' in row.get_text().strip():
-            cpu = row.get_text().replace('CPU', '').strip()
+            if 'CPU' in row.get_text().strip():
+                cpu = row.get_text().replace('CPU', '').strip()
 
-        if 'Вградена меморија' in row.get_text().strip():
-            rom_memory = row.get_text().replace('Вградена меморија', '').strip()
+            if 'Вградена меморија' in row.get_text().strip():
+                rom_memory = row.get_text().replace('Вградена меморија', '').strip()
 
-        if 'RAM меморија' in row.get_text().strip():
-            ram_memory = row.get_text().replace('RAM меморија', '').strip()
+            if 'RAM меморија' in row.get_text().strip():
+                ram_memory = row.get_text().replace('RAM меморија', '').strip()
 
-        if 'Батерија' in row.get_text().strip():
-            battery = row.get_text().replace('Батерија', '').strip()
+            if 'Батерија' in row.get_text().strip():
+                battery = row.get_text().replace('Батерија', '').strip()
 
-        if 'Предна камера' in row.get_text().strip():
-            front_camera = row.get_text().replace('Предна камера', '').strip()
+            if 'Предна камера' in row.get_text().strip():
+                front_camera = row.get_text().replace('Предна камера', '').strip()
 
-    new_offers.append(PhoneOffer(offer_shop, offer_name, price, ram_memory, rom_memory,
-                                 color, front_camera, back_camera, chipset, battery, operating_system, cpu, image_url,
-                                 offer_url, last_updated, is_validated, offer_description, offer_shop_code))
+        new_offers.append(PhoneOffer(offer_shop, offer_name, price, ram_memory, rom_memory,
+                                     color, front_camera, back_camera, chipset, battery, operating_system, cpu,
+                                     image_url,
+                                     offer_url, last_updated, is_validated, offer_description, offer_shop_code))
 
-for new_offer in new_offers:
-    flag = False
-    flag_price = False
-    offer_id = None
+    for new_offer in new_offers:
+        flag = False
+        flag_price = False
+        offer_id = None
+
+        for old_offer in database_offers:
+
+            if new_offer.offer_shop_code == old_offer.offer_shop_code:
+                flag = True
+                if new_offer.price != old_offer.price:
+                    flag_price = True
+                    offer_id = old_offer.offer_id
+
+        if flag:
+            # print('ALREADY IN DATABASE')
+            # print(new_offer)
+            # if it's already in database, check PRICE and if it's changed, change it !!!!!!
+            if flag_price:
+                print('PRICE CHANGED!')  # CHANGE PRICE
+                print('offer id: ' + str(offer_id))
+                headers = {'Content-type': 'application/json'}
+                requests.put(
+                    'http://localhost:8080/phoneoffer/' + str(offer_id) + '/changeprice/' + str(new_offer.price),
+                    headers=headers)
+        else:
+            print('ADDED')  # ADD OFFER
+            print(new_offer)
+            headers = {'Content-type': 'application/json'}
+            requests.post('http://localhost:8080/phoneoffer/addoffer', headers=headers,
+                          data=json.dumps(new_offer.__dict__,
+                                          default=str))
+
+    print('------------------------------------')
 
     for old_offer in database_offers:
+        flag = False
+        for new_offer in new_offers:
+            if old_offer.offer_shop_code == new_offer.offer_shop_code:
+                flag = True
 
-        if new_offer.offer_shop_code == old_offer.offer_shop_code:
-            flag = True
-            if new_offer.price != old_offer.price:
-                flag_price = True
-                offer_id = old_offer.offer_id
-
-    if flag:
-        # print('ALREADY IN DATABASE')
-        # print(new_offer)
-        # if it's already in database, check PRICE and if it's changed, change it !!!!!!
-        if flag_price:
-            print('PRICE CHANGED!')  # CHANGE PRICE
-            print('offer id: ' + str(offer_id))
-            headers = {'Content-type': 'application/json'}
-            requests.put('http://localhost:8080/phoneoffer/' + str(offer_id) + '/changeprice/' + str(new_offer.price),
-                         headers=headers)
-    else:
-        print('ADDED')  # ADD OFFER
-        print(new_offer)
-        headers = {'Content-type': 'application/json'}
-        requests.post('http://localhost:8080/phoneoffer/addoffer', headers=headers, data=json.dumps(new_offer.__dict__,
-                                                                                                    default=str))
-
-print('------------------------------------')
-
-for old_offer in database_offers:
-    flag = False
-    for new_offer in new_offers:
-        if old_offer.offer_shop_code == new_offer.offer_shop_code:
-            flag = True
-
-    if not flag:
-        print('OFFER DELETED')
-        print(old_offer)
-        # DELETE OFFER
-        requests.delete('http://localhost:8080/phoneoffer/deleteoffer/' + str(old_offer.offer_id))
+        if not flag:
+            print('OFFER DELETED')
+            print(old_offer)
+            # DELETE OFFER
+            requests.delete('http://localhost:8080/phoneoffer/deleteoffer/' + str(old_offer.offer_id))
+except Exception:
+    traceback.print_exc()
+    insert_script = 'INSERT INTO scrapper_info (store, recieved_at, status)' \
+                    ' VALUES (%s, %s, %s);'
+    insert_value = (offer_shop, last_updated, 'failed')
+    cur.execute(insert_script, insert_value)
+    db_connection.commit()
+    cur.close()
+    db_connection.close()
+else:
+    insert_script = 'INSERT INTO scrapper_info (store, recieved_at, status)' \
+                    ' VALUES (%s, %s, %s);'
+    insert_value = (offer_shop, last_updated, 'success')
+    cur.execute(insert_script, insert_value)
+    db_connection.commit()
+    cur.close()
+    db_connection.close()