import sys
import requests
class User:
    def __init__(self):
        self.access_token = None
        self.__appId = input('请输入您的appid:')
        self.__appSecret = input('请输入您的appsecret:')
        self.logintime = 3
        self.login()

    def login(self):
        while self.logintime:
            try:
                res = requests.get(
                    f"https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid={self.__appId}&secret={self.__appSecret}", )
                if res.status_code == 200:
                    self.access_token = res.json()['access_token']
                else:
                    print('输入错误!请尝试再次输入,还剩' + self.logintime + '次机会')
            except Exception as e:
                print('错误', res.json(), e)
            self.logintime = self.logintime - 1

    def quit(self):
        print('退出')
        sys.exit(0)

import os.path
import tqdm
import requests
from lxml import etree
import sys


class Article:
    def __init__(self):
        self.__url = None
        self.title = None
        self.__header = {
            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) '
                          'Chrome/108.0.0.0 Safari/537.36 Edg/108.0.1462.54'
        }
        self.filepathList = set()
        self.getUpdated()

    def getUpdated(self):
        url = 'https://www.pixivision.net/zh/c/illustration'
        res = requests.get(url,headers=self.__header)
        selector = etree.HTML(res.text)
        lastUpdatedNumber = selector.xpath("//li[@class='article-card-container'][1]/article/div[@class='arc__thumbnail-container']/a[1]/@href")[0]
        lastUpdatedNumber = lastUpdatedNumber.split('/')[-1]
        print('最新的图文号为',lastUpdatedNumber)

    def getarticle(self):
        self.__url = input('输入网站链接(仅限pixvision:https://www.pixivision.net/zh/a/number):')
        if self.__url.isnumeric():
            self.__url = 'https://www.pixivision.net/zh/a/' + self.__url
        try:
            res = requests.get(self.__url, headers=self.__header)
        except requests.exceptions.ProxyError as e:
            print('网络出现问题', e)
            sys.exit(-1)
            return
        except Exception as e:
            print('出错', e)
            sys.exit(-1)
            return
        selector = etree.HTML(res.text)
        imgLinks = selector.xpath(
            "////div[@class='_feature-article-body']/div[@class='article-item _feature-article-body__pixiv_illust']//div[@class='am__work__main']//img/@src")  # 返回为一列表
        self.title = selector.xpath("//h1[@class='am__title']/text()")
        self.download(imgLinks)

    def download(self, imgLinks):
        filepath = input('输入下载的目录路径(默认F:\公众号\图片):')

        if not os.path.exists(filepath):
            print('目录输入错误')
            return
        absfilepath = filepath + f'\\{self.title}'
        if os.path.exists(absfilepath):
            # 清空文件夹中的文件
            print('清空文件夹中的文件')
            del_list = os.listdir(absfilepath)
            for f in del_list:
                file_path = os.path.join(absfilepath, f)
                if os.path.isfile(file_path):
                    os.remove(file_path)
        else:
            print('生成文件夹')
            os.mkdir(absfilepath)
        self.filepathList.add(absfilepath)
        self.__header['Referer'] = 'https://www.pixiv.net/'
        pbar = tqdm.tqdm(imgLinks)
        total = len(imgLinks)
        for index, link in enumerate(pbar):
            filename = link.split('/')[-1]
            pbar.set_description("正在下载%s," % filename)
            pbar.set_postfix({'current': index + 1, 'total': total})
            try:
                res = requests.get(link, headers=self.__header)
                with open(absfilepath + f'\\{filename}', 'wb+') as f:
                    f.write(res.content)
            except Exception as e:
                print('需要设置代理', e)

import tqdm
import requests
import os


class upload:
    def __init__(self, ACCESS_TOKEN, FILEPATHLIST):
        self.url = f'https://api.weixin.qq.com/cgi-bin/material/add_material?access_token={ACCESS_TOKEN}&type' \
                   f'=image'
        self.filepath = FILEPATHLIST
        self.uploadPics()

    def uploadPics(self):

        if isinstance(self.filepath, str):
            self.uploadPic(self.filepath)
        else:
            for f in self.filepath:
                self.uploadPic(f)

    def uploadPic(self, filepath):
        if os.path.isdir(filepath):
            filelist = os.listdir(filepath)
            total = len(filelist)
            pbar = tqdm.tqdm(filelist)
            for index, f in enumerate(pbar):
                file_path = os.path.join(filepath, f)
                if os.path.isfile(file_path):
                    with open(file_path, 'rb') as file:
                        files = {
                            'media': file
                        }
                        res = requests.post(self.url, files=files)
                        pbar.set_description('正在上传' + res.json()['url'])
                        pbar.set_postfix({'current': index + 1, 'total': total})
        else:
            print('文件夹错误', filepath)

import time
from user import User
from getpic import Article
from uploadpic import upload

if __name__ == "__main__":
    uploadDownloadpic = '0'
    if input('直接上传图片?0表示直接上传,1表示下载\n') == '1':
        article = Article()
        while True:
            article.getarticle()
            doContinue = input('是否继续下载图片？1表示继续,0表示退出\n')
            if doContinue != '1':
                break
        uploadDownloadpic = input('是否上传刚才的图片?1表示是,0表示自己选择\n')
    u = User()
    if uploadDownloadpic == '1':
        upload(u.access_token, article.filepathList)
    else:
        filepath = input('输入需要上传的文件目录(单个):\n')
        upload(u.access_token, filepath)