ATRI/modules/b64/__init__.py


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

import os
import time
import base64
import requests
from pathlib import Path


path_temp_img = Path('.') / 'ATRI' / 'data' / 'temp' / 'img'


def b64_str_img_url(url: str):
    img_d = requests.get(url)
    name = 'temp.jpg'
    find_img = Path('.') / 'ATRI' / 'data' / 'temp' / 'img' / f'{name}'
    with open(find_img, 'wb') as f:
        f.write(img_d.content)
    if find_img.stat().st_size >= 1000000:
        print('超过1MB，跳过！')
        return
    else:
        with open(find_img, 'rb') as f:
            content = f.read()
        b64_str = base64.b64encode(content).decode()
        time.sleep(1)
        os.remove(str(path_temp_img) + '/' + name)
        return b64_str