Upload app.py with huggingface_hub

app.py

@@ -0,0 +1,110 @@
+import os
+import json
+import random
+import hashlib
+from flask import Flask, request
+
+def get_key_sha256(key):
+    sha256 = hashlib.sha256()
+    sha256.update(key.encode('utf-8'))
+    return sha256.hexdigest()
+
+match_sha256 = 'ffa007286cff75ab9c14e42282edd8e7e28b467a6e1bcd3d9b90dc40d375cae3'
+
+data_folder = 'jobs'
+stages = ['new', 'doing', 'done']
+for stage in stages:
+    os.makedirs(os.path.join(data_folder, stage), exist_ok=True)
+
+app = Flask(__name__)
+
+@app.route("/add", methods = ['POST'])
+def put():
+    d = request.json
+    key = d['key']
+    if not get_key_sha256(key) == match_sha256:
+        return 'err'
+    machine_type = d['machine_type']
+    job = d['job']
+    sha256 = get_key_sha256(job)
+    folder = os.path.join(data_folder, 'new', machine_type)
+    doing_folder = os.path.join(data_folder, 'doing', machine_type)
+    os.makedirs(folder, exist_ok=True)
+    os.makedirs(doing_folder, exist_ok=True)
+    file = os.path.join(folder, sha256)
+    open(file, 'w').write(job)
+    return 'ok'
+
+
+@app.route("/get", methods = ['GET'])
+def get():
+    d = request.json
+    key = d['key']
+    if not get_key_sha256(key) == match_sha256:
+        return 'err'
+    machine_type = d['machine_type']
+    folder = os.path.join(data_folder, 'new', machine_type)
+    if not os.path.isdir(folder):
+        return ''
+    doing_folder = os.path.join(data_folder, 'doing', machine_type)
+    files = os.listdir(folder)
+    if len(files) == 0:
+        return ''
+    file = random.choice(files)
+    source_file = os.path.join(folder, file)
+    target_file = os.path.join(doing_folder, file)
+    txt = open(source_file).read()
+    open(target_file, 'w').write(txt)
+    try:
+        os.remove(source_file)
+    except:
+        print('deleteerr')
+    return txt
+
+@app.route("/done", methods = ['GET'])
+def done():
+    d = request.json
+    key = d['key']
+    if not get_key_sha256(key) == match_sha256:
+        return 'err'
+    sha256 = d['sha256']
+    machine_type = d['machine_type']
+    file = os.path.join(data_folder, 'doing', machine_type, sha256)
+    try:
+        os.remove(file)
+    except:
+        return 'derr'
+    return 'ok'
+
+
+def get_folder_list(folder, full=False):
+    machine_types = os.listdir(folder)
+    d = {}
+    for machine_type in machine_types:
+        fs = os.listdir(os.path.join(folder, machine_type))
+        if full:
+            dd = {}
+            for f in fs:
+                dd[f] = open(os.path.join(folder, machine_type, f)).read()
+            d[machine_type] = dd
+        else:
+            
+            d[machine_type] = fs
+    return d
+
+@app.route("/stat", methods = ['GET'])
+def stat():
+    d = request.json
+    key = d['key']
+    if not get_key_sha256(key) == match_sha256:
+        return 'err'
+    full = int(d['full'])
+    stat = {}
+    stat['new'] = get_folder_list(os.path.join(data_folder, 'new'), full)
+    stat['doing'] = get_folder_list(os.path.join(data_folder, 'doing'), full)
+    return json.dumps(stat) 
+
+
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860, debug=True)