mirror of
https://github.com/simon987/music-graph-scripts.git
synced 2025-04-10 05:56:42 +00:00
57 lines
1.2 KiB
Python
57 lines
1.2 KiB
Python
import json
|
|
from multiprocessing.pool import ThreadPool
|
|
|
|
from task_tracker_drone.src.tt_drone.api import TaskTrackerApi, Worker
|
|
|
|
TT_API_URL = "https://tt.simon987.net/api"
|
|
TT_PROJECT = 5
|
|
|
|
|
|
done = set()
|
|
# with sqlite3.connect(sys.argv[1]) as conn:
|
|
# cur = conn.cursor()
|
|
# cur.execute("SELECT id FROM covers")
|
|
# for mbid in cur.fetchall():
|
|
# done.add(mbid[0])
|
|
|
|
api = TaskTrackerApi(TT_API_URL)
|
|
|
|
worker = Worker.from_file(api)
|
|
if not worker:
|
|
worker = api.make_worker("caa scraper")
|
|
worker.dump_to_file()
|
|
worker.request_access(TT_PROJECT, True, True)
|
|
input("Give permission to " + worker.alias)
|
|
|
|
|
|
def mktask(mbids):
|
|
res = worker.submit_task(
|
|
project=TT_PROJECT,
|
|
recipe=json.dumps(mbids),
|
|
hash64=hash(mbids[0]),
|
|
max_assign_time=60 * 30,
|
|
priority=1,
|
|
unique_str=None,
|
|
verification_count=None,
|
|
max_retries=5,
|
|
)
|
|
print(res.text)
|
|
|
|
|
|
def lines():
|
|
with open("in/release") as f:
|
|
buf = list()
|
|
|
|
for line in f:
|
|
cols = line.split("\t")
|
|
|
|
buf.append(cols[1])
|
|
if len(buf) == 75:
|
|
a = list(buf)
|
|
buf.clear()
|
|
yield a
|
|
|
|
|
|
pool = ThreadPool(processes=20)
|
|
pool.map(func=mktask, iterable=lines())
|