求好心人帮忙改一份python源码

huangxiaolei

求大神帮忙改成服务端可以api调用，  取收件箱，垃圾箱传参数进去，比如传参数1是取收件箱2是垃圾箱这样。

wer3255561

import email
import email.header
import requests
import imaplib
import sys
# 1. 使用refresh_token获取access_token
def get_accesstoken(refresh_token,client_id):
    data = {
        'client_id': client_id,
        'grant_type': 'refresh_token',
        'refresh_token': refresh_token
    }
    ret = requests.post('https://login.microsoftonline.com/consumers/oauth2/v2.0/token', data=data)
    print(ret.text)
    return ret.json()['access_token']


def generate_auth_string(user, token):
    auth_string = f"user={user}\1auth=Bearer {token}\1\1"
    return auth_string


def tuple_to_str(tuple_):
    """
    元组转为字符串输出
    :param tuple_: 转换前的元组，QQ邮箱格式为(b'\xcd\xf5\xd4\xc6', 'gbk')或者(b' <XXXX@163.com>', None)，163邮箱格式为('<XXXX@163.com>', None)
    :return: 转换后的字符串
    """
    if tuple_[1]:
        out_str = tuple_[0].decode(tuple_[1])
    else:
        if isinstance(tuple_[0], bytes):
            out_str = tuple_[0].decode('gbk')
        else:
            out_str = tuple_[0]
    return out_str


def fetch_email_body(mail, item):
    try:
        ret, data = mail.fetch(item, '(RFC822)')
        msg = email.message_from_string(data[0][1].decode('utf-8'))
        sub = msg.get('subject')
        sub_text = email.header.decode_header(str(sub))
        # 主题
        sub_detail = ''
        if sub_text[0]:
            sub_detail = tuple_to_str(sub_text[0])
        print(sub_detail)
        for part in msg.walk():
            if not part.is_multipart():
                content_type = part.get_content_type()
                name = part.get_filename()
                if not name:
                    txt = str(part.get_payload(decode=True))
                    if content_type == 'text/html':
                        # mail.store(item, '-FLAGS', '\\SEEN')  设为已读
                        print(txt)
    except Exception as e:
        print(e)


def getmail(sel, mail):
    mail.select(sel)
    status, messages = mail.search(None, 'ALL') #UNSEEN 为未读邮件
    all_emails = messages[0].split()
    print(all_emails)
    for item in all_emails:
        fetch_email_body(mail, item)


# 2. 使用访问令牌连接 IMAP
def connect_imap(emailadr, access_token):
    mail = imaplib.IMAP4_SSL('outlook.live.com')
    print(generate_auth_string(emailadr, access_token))
    mail.authenticate('XOAUTH2', lambda x: generate_auth_string(emailadr, access_token))
    getmail('INBOX', mail) #读取收件箱
    getmail('Junk', mail) #读取垃圾箱
    # 关闭连接
    mail.logout()


# 主流程
if __name__ == '__main__':

    emailadr = sys.argv[1]
    refresh_token = sys.argv[2]
    access_token = get_accesstoken(refresh_token,sys.argv[3])
    print("获取的访问令牌:", access_token)
    # 连接 IMAP
    connect_imap(emailadr, access_token)

huangxiaolei

令牌跟id也是要传参数进去的

fire9

用flask呗                 

黑马科技

珂珂珂珂

人杰啊

10精币都懒得打开gpt帮你生成

huangxiaolei

人杰啊 发表于 2024-10-8 21:13
10精币都懒得打开gpt帮你生成

啊 随便设置的，，，，

huangxiaolei

fire9 发表于 2024-10-8 19:14
用flask呗

不懂啊

wer3255561

[Python] 纯文本查看 复制代码

import email
import email.header
import requests
import imaplib
import sys
# 1. 使用refresh_token获取access_token
def get_accesstoken(refresh_token,client_id):
    data = {
        'client_id': client_id,
        'grant_type': 'refresh_token',
        'refresh_token': refresh_token
    }
    ret = requests.post('https://login.microsoftonline.com/consumers/oauth2/v2.0/token', data=data)
    print(ret.text)
    return ret.json()['access_token']


def generate_auth_string(user, token):
    auth_string = f"user={user}\1auth=Bearer {token}\1\1"
    return auth_string


def tuple_to_str(tuple_):
    """
    元组转为字符串输出
    :param tuple_: 转换前的元组，QQ邮箱格式为(b'\xcd\xf5\xd4\xc6', 'gbk')或者(b' <XXXX@163.com>', None)，163邮箱格式为('<XXXX@163.com>', None)
    :return: 转换后的字符串
    """
    if tuple_[1]:
        out_str = tuple_[0].decode(tuple_[1])
    else:
        if isinstance(tuple_[0], bytes):
            out_str = tuple_[0].decode('gbk')
        else:
            out_str = tuple_[0]
    return out_str


def fetch_email_body(mail, item):
    try:
        ret, data = mail.fetch(item, '(RFC822)')
        msg = email.message_from_string(data[0][1].decode('utf-8'))
        sub = msg.get('subject')
        sub_text = email.header.decode_header(str(sub))
        # 主题
        sub_detail = ''
        if sub_text[0]:
            sub_detail = tuple_to_str(sub_text[0])
        print(sub_detail)
        for part in msg.walk():
            if not part.is_multipart():
                content_type = part.get_content_type()
                name = part.get_filename()
                if not name:
                    txt = str(part.get_payload(decode=True))
                    if content_type == 'text/html':
                        # mail.store(item, '-FLAGS', '\\SEEN')  设为已读
                        print(txt)
    except Exception as e:
        print(e)


def getmail(sel, mail):
    mail.select(sel)
    status, messages = mail.search(None, 'ALL') #UNSEEN 为未读邮件
    all_emails = messages[0].split()
    print(all_emails)
    for item in all_emails:
        fetch_email_body(mail, item)


# 2. 使用访问令牌连接 IMAP
def connect_imap(emailadr, access_token,sf):
    mail = imaplib.IMAP4_SSL('outlook.live.com')
    print(generate_auth_string(emailadr, access_token))
    mail.authenticate('XOAUTH2', lambda x: generate_auth_string(emailadr, access_token))
    if number == 1:
        getmail('INBOX', mail) #读取收件箱
    else:
        getmail('Junk', mail) #读取垃圾箱
    # 关闭连接
    mail.logout()


# 主流程
if __name__ == '__main__':

    emailadr = sys.argv[1]
    refresh_token = sys.argv[2]
    access_token = get_accesstoken(refresh_token,sys.argv[3])
    print("获取的访问令牌:", access_token)
    # 连接 IMAP
    connect_imap(emailadr, access_token,sys.argv[4])

Huaz

wer3255561 发表于 2024-10-16 00:49
[mw_shl_code=python,true]import email
import email.header
import requests

你不得加点精币 加点我给你改