5m4ck3r commited on
Commit
b2b8786
1 Parent(s): d8df835

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +7 -9
main.py CHANGED
@@ -95,10 +95,9 @@ def gettans_task():
95
  file_content.name = file.filename
96
  else:
97
  rawFile = request.data
98
- file = io.BytesIO(rawFile)
99
- if not request.args.get("name"):
100
- filename = ''.join(random.choice(string.ascii_letters + string.digits) for _ in range(10)) + ".png"
101
- file.name = request.args.get("name", filename)
102
 
103
  headers = dict(request.headers)
104
  if not headers.get("KEY") != os.getenv("KEY"):
@@ -113,7 +112,7 @@ def task_ocr():
113
  if len(request.data) > 0:
114
  pass
115
  else:
116
- return jsonify({"status" : False, "msg" : "No file found", "len" : len(request.data), "form" : dict(request.form)}), 400
117
 
118
  if 'file' in request.files:
119
  file = request.files['file']
@@ -123,10 +122,9 @@ def task_ocr():
123
  file_content.name = file.filename
124
  else:
125
  rawFile = request.data
126
- file = io.BytesIO(rawFile)
127
- if not request.args.get("name"):
128
- filename = ''.join(random.choice(string.ascii_letters + string.digits) for _ in range(10)) + ".png"
129
- file.name = request.args.get("name", filename)
130
 
131
  headers = dict(request.headers)
132
  if not headers.get("KEY") != os.getenv("KEY"):
 
95
  file_content.name = file.filename
96
  else:
97
  rawFile = request.data
98
+ file_content = io.BytesIO(rawFile)
99
+ filename = ''.join(random.choice(string.ascii_letters + string.digits) for _ in range(10)) + ".png"
100
+ file_content.name = request.args.get("name", filename)
 
101
 
102
  headers = dict(request.headers)
103
  if not headers.get("KEY") != os.getenv("KEY"):
 
112
  if len(request.data) > 0:
113
  pass
114
  else:
115
+ return jsonify({"status" : False, "msg" : "No file found"}), 400
116
 
117
  if 'file' in request.files:
118
  file = request.files['file']
 
122
  file_content.name = file.filename
123
  else:
124
  rawFile = request.data
125
+ file_content = io.BytesIO(rawFile)
126
+ filename = ''.join(random.choice(string.ascii_letters + string.digits) for _ in range(10)) + ".png"
127
+ file_content.name = request.args.get("name", filename)
 
128
 
129
  headers = dict(request.headers)
130
  if not headers.get("KEY") != os.getenv("KEY"):