Hell13Cat
233 lines
8.3 KiB
Python
233 lines
8.3 KiB
Python
import traceback
|
|
from pygelbooru import Gelbooru
|
|
import PySimpleGUI as sg
|
|
import PIL
|
|
from PIL import Image
|
|
import requests
|
|
import json
|
|
import progressbar
|
|
|
|
def get_cfg(name):
|
|
datacfg = json.load(open("config.json", "r", encoding='utf-8'))
|
|
return datacfg[name]
|
|
|
|
def watch(filename):
|
|
img = Image.open(filename)
|
|
width = 600
|
|
ratio = (width / float(img.size[0]))
|
|
height = int((float(img.size[1]) * float(ratio)))
|
|
img = img.resize((width, height), PIL.Image.ANTIALIAS)
|
|
size = (width, height)
|
|
img.save("tmp.png")
|
|
#im = im.resize(size, resample=Image.BICUBIC)
|
|
sg.theme('DarkPurple6')
|
|
layout = [
|
|
[sg.Text("Изображение сохранено")],
|
|
[sg.Image("tmp.png", size=size, key='-IMAGE-')],
|
|
]
|
|
window = sg.Window('Window Title', layout, margins=(0, 0), finalize=True)
|
|
|
|
async def get1():
|
|
gelbooru = Gelbooru(get_cfg("api_key"), get_cfg("693972"))
|
|
list_tags = await gelbooru.tag_list(name='dog ears')
|
|
print(list_tags)
|
|
list_tags = await gelbooru.tag_list(name_pattern='%small%', limit=4)
|
|
print(list_tags)
|
|
result = await gelbooru.random_post(tags=['cat ears', '1girl', 'cat hood', 'bell'], exclude_tags=['nude'])
|
|
print(result)
|
|
req = requests.get(str(result))
|
|
cont = req.content
|
|
open(result.filename, "wb").write(cont)
|
|
results = await gelbooru.search_posts(tags=['dog ears', '1girl'], exclude_tags=['nude'])
|
|
print(results)
|
|
|
|
async def gettest():
|
|
watch("testimg.jpeg")
|
|
fff = input("ddd")
|
|
|
|
async def get2():
|
|
gelbooru = Gelbooru(get_cfg("api_key"), get_cfg("693972"))
|
|
list_tags = await gelbooru.tag_list(name_pattern='%', limit=10000)
|
|
print(len(list_tags))
|
|
list_tagss = []
|
|
for ii in list_tags:
|
|
list_tagss.append(str(ii))
|
|
list_tagss.sort()
|
|
text = "\n".join(list_tagss)
|
|
open("tags.txt", "w").write(text)
|
|
|
|
|
|
#loop = asyncio.get_event_loop()
|
|
#loop.run_until_complete(get1())
|
|
|
|
def id_ready(id):
|
|
dataf = json.load(open("data_info.json", "r", encoding='utf-8'))
|
|
ids = dataf["id_set"]
|
|
if id in ids:
|
|
return 1
|
|
else:
|
|
return 0
|
|
|
|
def get_count():
|
|
dataf = json.load(open("data_info.json", "r", encoding='utf-8'))
|
|
ids = dataf["id_set"]
|
|
return len(ids)
|
|
|
|
def add_data(data):
|
|
resp = {}
|
|
try:
|
|
dataf = json.load(open("data_info.json", "r", encoding='utf-8'))
|
|
ids = dataf["id_set"]
|
|
ids.append(data["id"])
|
|
dataf["id_set"] = ids
|
|
datas = dataf["data_set"]
|
|
tags = data["tags"].split(" ")
|
|
dataa = {
|
|
"id": data["id"],
|
|
"file": data["image"],
|
|
"data": data["created_at"],
|
|
"rating": data["rating"],
|
|
"source": data["source"],
|
|
"tags": tags
|
|
}
|
|
datas.append(dataa)
|
|
dataf["data_set"] = datas
|
|
tagss = dataf["tags"]
|
|
for ii in tags:
|
|
if ii in tagss:
|
|
pass
|
|
else:
|
|
tagss.append(ii)
|
|
dataf["tags"] = tagss
|
|
json.dump(dataf, open("data_info.json", "w", encoding='utf-8'), ensure_ascii=False, indent=4)
|
|
resp["code"] = 200
|
|
resp["text"] = "OK"
|
|
except BaseException as e:
|
|
resp["code"] = 602
|
|
resp["text"] = str(e)
|
|
resp["fulltext"] = str(traceback.format_exc)
|
|
return resp
|
|
|
|
def save_data(url, filename):
|
|
resp = {}
|
|
try:
|
|
req = requests.get(url, stream=True)
|
|
if req.status_code == 200:
|
|
f = open("data_img/"+filename, 'wb')
|
|
file_size = int(req.headers['Content-Length'])
|
|
chunk = 1
|
|
num_bars = file_size / chunk
|
|
bar = progressbar.ProgressBar(maxval=num_bars)
|
|
bar.start()
|
|
i = 0
|
|
for chunk in req.iter_content():
|
|
f.write(chunk)
|
|
bar.update(i)
|
|
i+=1
|
|
bar.update(num_bars)
|
|
bar.finish()
|
|
f.close()
|
|
resp["code"] = req.status_code
|
|
resp["text"] = req.reason
|
|
except BaseException as e:
|
|
resp["code"] = 601
|
|
resp["text"] = str(e)
|
|
resp["fulltext"] = str(traceback.format_exc)
|
|
return resp
|
|
|
|
|
|
def get_data(tags, limit, pid):
|
|
try:
|
|
tags = tags.replace(" ", "+")
|
|
append_auth = "&api_key="+get_cfg("api_key")+"&user_id="+get_cfg("user_id")
|
|
base_root = "https://gelbooru.com/index.php?"
|
|
append_data = "page=dapi&s=post&q=index&json=1&limit={limit}&tags={tags}&pid={pid}".format(pid=str(pid), tags=tags, limit=str(limit))
|
|
url_res = base_root + append_data + append_auth
|
|
req = requests.get(url_res)
|
|
if req.status_code == 200:
|
|
data = req.json()
|
|
else:
|
|
data = {}
|
|
data["code"] = req.status_code
|
|
data["text"] = req.reason
|
|
return data
|
|
except BaseException as e:
|
|
resp = {}
|
|
resp["code"] = 600
|
|
resp["text"] = str(e)
|
|
resp["fulltext"] = str(traceback.format_exc)
|
|
return resp
|
|
|
|
|
|
def download(datain):
|
|
print(datain)
|
|
data_search = json.load(open("search_data.json", "r", encoding='utf-8'))
|
|
animated_tag = ["", " -animated", " webm", " animated", " webm"]
|
|
rating_tag = ["", " rating:general", " rating:questionable ", " rating:explicit",]
|
|
pages = data_search["pages"]
|
|
limits = data_search["limits"]
|
|
animated = data_search["animated"]
|
|
rating = data_search["rating"]
|
|
c_total_dw = 0
|
|
c_total_ck = 0
|
|
c_total_er = 0
|
|
tags = datain["tags"]
|
|
print("[Request] ", tags, "\nAnimated type:", animated, "\nPages:", pages, "\nLimits on page:", limits)
|
|
for page in range(0, pages):
|
|
data = get_data(tags+animated_tag[animated]+rating_tag[rating], limits, page)
|
|
if data["code"] != 200:
|
|
print("[Error] Code:", data["code"], "\nDescription:", data["text"])
|
|
if data["code"] >= 600:
|
|
print("----Full info----\n", data["fulltext"], "\n----Full info----")
|
|
else:
|
|
if "post" in data.keys():
|
|
data = data["post"]
|
|
count = 1
|
|
c_page_dw = 0
|
|
c_page_ck = 0
|
|
c_page_er = 0
|
|
for one_post in data:
|
|
print("[", tags, "|", animated, "]-[", page+1 , "/", pages, "]-[", count, "/", len(data), "]", one_post["id"], "-", one_post["image"])
|
|
count += 1
|
|
if id_ready(one_post["id"]) == 0:
|
|
resp = save_data(one_post["file_url"], one_post["image"])
|
|
if resp["code"] == 200:
|
|
resp = add_data(one_post)
|
|
if resp["code"] != 200:
|
|
print("[Error] Skiped\nCode:", resp["code"], "\nDescription:", resp["text"])
|
|
if data["code"] >= 600:
|
|
print("----Full info----\n", data["fulltext"], "\n----Full info----")
|
|
else:
|
|
c_total_dw += 1
|
|
c_page_dw += 1
|
|
print("[Notify] Saved")
|
|
else:
|
|
print("[Error] Skiped\nCode:", resp["code"], "\nDescription:", resp["text"])
|
|
if data["code"] >= 600:
|
|
print("----Full info----\n", data["fulltext"], "\n----Full info----")
|
|
c_total_er += 1
|
|
c_page_er += 1
|
|
else:
|
|
c_total_ck += 1
|
|
c_page_ck += 1
|
|
print("[Notify] Skiped")
|
|
print("[", page+1 , "/", pages, "] Current count images:", get_count(), "\nDownloads:", c_page_dw, "-", c_total_dw, "\nSkiped:", c_page_ck, "-", c_total_ck, "\nError:", c_page_er, "-", c_total_er)
|
|
else:
|
|
print("[Notify] Skiped page")
|
|
|
|
data_send = {"tags":"", "rating":0, "animated":0}
|
|
animated_tag = ["", " -animated", " webm", " animated", " webm"]
|
|
rating_tag = ["", " rating:general", " rating:questionable ", " rating:explicit",]
|
|
|
|
|
|
inp_tag = input(">>> ")
|
|
if inp_tag == "":
|
|
list_dw_tags = open("list_tag.txt", "r").read().split("\n")
|
|
page = 1
|
|
for fff in list_dw_tags:
|
|
inp_tag = fff
|
|
print("[", page , "/", len(list_dw_tags), "] Current dw tags: ", fff)
|
|
page += 1
|
|
download({"tags":inp_tag})
|
|
else:
|
|
download({"tags":inp_tag}) |