microbamboo commited on
Commit
16e09f6
·
verified ·
1 Parent(s): 0796c8d

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +13 -28
app.py CHANGED
@@ -1,43 +1,28 @@
1
  import os
2
- import hashlib
3
  from flask import Flask, request
4
- import subprocess
5
-
6
- def get_key_sha256(key):
7
- sha256 = hashlib.sha256()
8
- sha256.update(key.encode('utf-8'))
9
- return sha256.hexdigest()
10
-
11
- match_sha256 = 'ffa007286cff75ab9c14e42282edd8e7e28b467a6e1bcd3d9b90dc40d375cae3'
12
 
13
  data_folder = 'static'
14
- if not os.path.isdir(data_folder):
15
- os.makedirs(data_folder)
16
 
17
  app = Flask(__name__, static_folder=data_folder, static_url_path='')
18
 
19
-
20
- @app.route("/put", methods = ['POST'])
21
  def put():
22
  d = request.headers
23
- key = d['Key']
24
  file_name = d['File']
25
- if not get_key_sha256(key) == match_sha256:
26
- return 'err'
27
- file = file_name
28
  folder = os.path.split(file)[0]
29
- if len(folder) > 0:
30
- os.makedirs(folder, exist_ok=True)
31
  f = request.files['file']
32
  f.save(file)
33
- if file_name == 'all.tar.gz':
34
- os.system('tar zfx all.tar.gz')
35
- o = subprocess.getoutput('python main.py')
36
- else:
37
- o = subprocess.getoutput(f'python {file}')
38
- os.system('rm -rf ./*')
39
- return o
40
 
41
- if __name__ == "__main__":
42
- app.run(host="0.0.0.0", port=7860, debug=False)
 
 
 
 
43
 
 
 
 
1
  import os
 
2
  from flask import Flask, request
 
 
 
 
 
 
 
 
3
 
4
  data_folder = 'static'
5
+ os.makedirs(data_folder, exist_ok=True)
 
6
 
7
  app = Flask(__name__, static_folder=data_folder, static_url_path='')
8
 
9
+ @app.route("/cut", methods = ['POST'])
 
10
  def put():
11
  d = request.headers
 
12
  file_name = d['File']
13
+ file = os.path.join(data_folder, file_name)
 
 
14
  folder = os.path.split(file)[0]
15
+ os.makedirs(folder, exist_ok=True)
 
16
  f = request.files['file']
17
  f.save(file)
18
+ return "ok"
 
 
 
 
 
 
19
 
20
+ @app.route("/c")
21
+ def c():
22
+ import shutil
23
+ shutil.rmtree(data_folder)
24
+ os.makedirs(data_folder)
25
+ return "ok"
26
 
27
+ if __name__ == "__main__":
28
+ app.run(host="0.0.0.0", port=7860, debug=True)