Final_work/main.py at main · IlinSergey/Final_work · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
import requests
import config
import json
import os
from tqdm import tqdm


class VkAgent:
    def __init__(self, token: str, id: str):
        self.token = token
        self.id = id

    def get_response(self, url, params):
        return requests.get(url, params=params).json()

    def get_link(self, response, i=int):
        return response['response']['items'][i]['sizes'][-1]['url']

    def get_photo(self, count: int):

        url = 'https://api.vk.com/method/photos.get'
        params = {
            'owner_id': self.id,
            'album_id': 'profile',
            'extended': '1',
            'photo_sizes': '1',
            'rev': '1',
            'access_token': self.token,
            'v': '5.131'
        }
        photo_list = []
        response = self.get_response(url, params)
        for i in range(count):
            file_name = response['response']['items'][i]['likes']['count']
            link = self.get_link(response, i)
            photo_info = {}
            photo_info['file_name'] = file_name
            photo_info['size'] = response['response']['items'][i]['sizes'][-1]['type']
            photo_list.append(photo_info)
            f = open(f'backup\{file_name}.jpg', 'wb')
            ufr = requests.get(link)
            f.write(ufr.content)
            f.close()
        with open(r'backup\files_log.json', 'w') as f:
            f.write(json.dumps(photo_list))
        print('Фото скачаны')

class YaUploader:
    files_url = 'https://cloud-api.yandex.net/v1/disk/resources/files'
    upload_url = 'https://cloud-api.yandex.net/v1/disk/resources'
    def __init__(self, token: str):
        self.token = token

    def get_upload_link(self, file, replace = True):
        headers = {'Content-Type': 'application/json', 'Accept': 'application/json', 'Authorization': f'OAuth {self.token}'}
        response = requests.get(f'{self.upload_url}/upload?path=backup/{file}&overwrite={replace}', headers=headers).json()
        href = response['href']
        if not href:
            return print(f'Ошибка, ссылка не получена! {response["message"]}')
        return href
    def upload(self, path: str, file: str):
        href = self.get_upload_link(file)
        with open(path, 'rb') as file:
            try:
                requests.put(href, files={'file':file})
            except KeyError:
                print(f'Файл не загружен, ошибка: {self.get_upload_link(path)["message"]}')

    def vk_photo_backup(self, cout_photo: int):
        vk.get_photo(cout_photo)
        file_list = os.listdir(r'backup')
        for file in tqdm(file_list):
            path = f'backup\\{file}'
            uploader.upload(path, file)
        print('Файлы успешно загружены')


uploader = YaUploader(config.yandex_token)
vk = VkAgent(config.vk_token, '11606581')

uploader.vk_photo_backup(5)