v1.1 Первая версия пакетика

Author: DrillCoder

__init__.py

@@ -0,0 +1,12 @@
+from python_pik_api_wrapper.entity.entity import Entity
+
+
+class Block(Entity):
+    def __init__(self, id: str, name: str, sort: int):
+        super().__init__(id)
+        if type(name) is not str:
+            raise ValueError
+        self.name = name
+        if type(sort) is not int:
+            raise ValueError
+        self.sort = sort

python_pik_api_wrapper/__init__.py

@@ -0,0 +1,15 @@
+from python_pik_api_wrapper.entity.entity import Entity
+
+
+class Bulk(Entity):
+    def __init__(self, id: str, name: str, sort: int, type_id: int):
+        super().__init__(id)
+        if type(name) is not str:
+            raise ValueError
+        self.name = name
+        if type(sort) is not int:
+            raise ValueError
+        self.sort = sort
+        if type(type_id) is not int:
+            raise ValueError
+        self.type_id = type_id

python_pik_api_wrapper/entity/__init__.py

@@ -0,0 +1,17 @@
+class Entity:
+    def __init__(self, id: str):
+        if type(id) is not str:
+            raise ValueError
+        self.id = id
+
+    def __repr__(self):
+        str_list = []
+        for item in self.__dict__:
+            if type(getattr(self, item)) is list:
+                elements = []
+                for element in getattr(self, item):
+                    elements.append(f'{element.__repr__()}')
+                str_list.append(f'{item}:\n' + "\n".join(elements))
+                continue
+            str_list.append(f'{item}: {getattr(self, item)}')
+        return ', '.join(str_list)

python_pik_api_wrapper/entity/block.py

@@ -0,0 +1,18 @@
+from typing import List
+
+from python_pik_api_wrapper.entity.entity import Entity
+from python_pik_api_wrapper.value_object.photo import Photo
+
+
+class Item(Entity):
+    def __init__(self, id: str, public_date: str, date: str, photos: List[Photo]):
+        super().__init__(id)
+        if type(public_date) is not str:
+            raise ValueError
+        self.public_date = public_date
+        if type(date) is not str:
+            raise ValueError
+        self.date = date
+        if type(photos) is not list:
+            raise ValueError
+        self.photos = photos

python_pik_api_wrapper/entity/bulk.py

@@ -0,0 +1,12 @@
+from python_pik_api_wrapper.entity.entity import Entity
+
+
+class Location(Entity):
+    def __init__(self, id: str, name: str, sort: int):
+        super().__init__(id)
+        if type(name) is not str:
+            raise ValueError
+        self.name = name
+        if type(sort) is not int:
+            raise ValueError
+        self.sort = sort

python_pik_api_wrapper/entity/entity.py

@@ -0,0 +1,95 @@
+from typing import List
+
+import requests
+
+from python_pik_api_wrapper.entity.block import Block
+from python_pik_api_wrapper.entity.bulk import Bulk
+from python_pik_api_wrapper.entity.item import Item
+from python_pik_api_wrapper.entity.location import Location
+from python_pik_api_wrapper.value_object.photo import Photo
+
+
+class PikWrapper:
+    def __init__(self):
+        self.base_url = 'https://api.pik.ru/v1/'
+        self.headers = {
+            'accept': '*/*',
+            'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_1) AppleWebKit/605.1.15 (KHTML, like Gecko) '
+                          'Version/13.0.3 Safari/605.1.15',
+            'wrapper': 'https://github.com/drillcoder/python_pik_api_wrapper'
+        }
+
+    def send_request(self, url:str) -> List:
+        if type(url) is not str:
+            raise ValueError
+        response = requests.get(self.base_url + url, headers=self.headers)
+        if response.status_code != requests.codes.ok:
+            raise ValueError
+        if response.status_code == requests.codes.unprocessable:
+            return []
+        content = response.json()
+        if type(content) is dict:
+            if content.get('message') == 'ERR_NEWS_NOT_FOUND':
+                raise []
+            if type(content.get('items')) is not None:
+                content = content.get('items')
+        if type(content) is not list:
+            raise ValueError
+        return content
+
+    def get_locations(self) -> List[Location]:
+        locations = self.send_request('location')
+        locations_list = []
+        count = 0
+        for location in locations:
+            if location['id'] == 2:
+                locations_list.append(Location('2', 'Москва', count))
+                count += 1
+                locations_list.append(Location('3', 'Московская область', count))
+                continue
+            locations_list.append(Location(str(location['id']), location['name'], count))
+            count += 1
+        return locations_list
+
+    def get_blocks(self, location: Location) -> List[Block]:
+        if type(location) is not Location:
+            raise ValueError
+        blocks = self.send_request(f"block?metadata=1&types=1,2&locations={location.id}")
+        blocks_list = []
+        for block in blocks:
+            blocks_list.append(Block(str(block['id']), block['name'], block['sort']))
+        return blocks_list
+
+    def get_bulks(self, block: Block) -> List[Bulk]:
+        if type(block) is not Block:
+            raise ValueError
+        bulks = self.send_request(f"bulk?block_id={block.id}")
+        bulks_list = []
+        count = 0
+        for bulk in bulks:
+            if bulk['type_id'] not in {1, 2, 103} or len(bulk['name']) == 0:
+                continue
+            bulks_list.append(Bulk(str(bulk['id']), bulk['name'], count, bulk['type_id']))
+            count += 1
+        return bulks_list
+
+    def get_items(self, bulk: Bulk) -> List[Item]:
+        if type(bulk) is not Bulk:
+            raise ValueError
+        items = self.send_request(f"news?limit=all&is_content=1&is_progress=1&bulk_id={bulk.id}")
+        items_list = []
+        for item in items:
+            url = 'https:' + item['preview']
+            images_path = [item['preview']]
+            photos = [Photo(url, 0)]
+            sort = 1
+            if type(item.get('gallery')) is list:
+                for photo in item['gallery']:
+                    if photo['file_path'] in images_path:
+                        continue
+                    images_path.append(photo['file_path'])
+                    url = 'https:' + photo['file_path']
+                    photos.append(Photo(url, sort))
+                    sort += 1
+            items_list.append(Item(item['id'], item['public_date'], item['date'], photos))
+        return items_list