Refactoring for wrapper

common/tinderwrapper.py

@@ -1,5 +1,4 @@
 import requests
-import json
 
 class TinderWrapper():
   def __init__(self, token):
@@ -28,37 +27,39 @@ class TinderWrapper():
 
   def get_updates(self, last_activity_date):
     try:
-      url = self.host + '/updates'
-      r = requests.post(url, headers=self.headers, data=json.dumps({"last_activity_date": last_activity_date}))
-      print(r.json())
-      return r.json()
+        url = f"{self.host}/updates"  # Using f-string for formatting
+        payload = {"last_activity_date": last_activity_date}
+        response = requests.post(url, headers=self.headers, json=payload)  # Using json parameter instead of manually converting to JSON string
+        response.raise_for_status()  # Raises an exception for 4xx or 5xx status codes
+        return response.json()
     except requests.exceptions.RequestException as e:
-      print(e)
+        print(e)
 
   def get_photos(self, person):
-    photos = person['photos']
-    photo_urls = []
-    for photo in photos:
-      photo_urls.append(photo['url'])
-    return photo_urls
+    return [photo['url'] for photo in person['photos']]
 
   def create_dump(self, update_progress, max_percent=50, last_activity_date='1997-03-25T22:49:41.151Z'):
-    update_progress(0, 'Делаю запрос в Tinder...')
+    update_progress(0, 'Выполняю запрос в Tinder...')
     updates = self.get_updates(last_activity_date)['matches']
     update_progress(0, 'Обрабатываю ответ Tinder...')
+    
     output = {}
+    total_updates = len(updates)
+
     for i, update in enumerate(updates):
-      person = update['person']
-      photos = self.get_photos(person)
-      person_id = person['_id']
-      name = person['name']
-      messages = update['messages']
-      output[person_id] = {
-          'name': name,
-          'messages': messages,
-          'photos': photos
-      }
-      percent = ((i + 1) / len(updates)) * max_percent / 100
-      update_progress(percent, 'Обрабатываю ответ Tinder...')
-    
+        person = update['person']
+        photos = self.get_photos(person)
+        person_id = person['_id']
+        name = person['name']
+        messages = update['messages']
+        
+        output[person_id] = {
+            'name': name,
+            'messages': messages,
+            'photos': photos
+        }
+        
+        percent = ((i + 1) / total_updates) * max_percent
+        update_progress(percent, 'Обрабатываю ответ Tinder...')
+
     return output