mirror of
https://github.com/makefu/bump.git
synced 2024-03-22 11:32:52 +01:00
code is now working with py3
This commit is contained in:
parent
63f233e0b9
commit
1537273b09
1 changed files with 20 additions and 13 deletions
27
bump/init.py
27
bump/init.py
|
@ -1,9 +1,10 @@
|
|||
from flask import Flask, Response, request
|
||||
from functools import wraps
|
||||
#import sys
|
||||
import sys
|
||||
import random
|
||||
import string
|
||||
storedir = "store"
|
||||
import os
|
||||
storedir = "%s/../store" % os.path.dirname(os.path.abspath(sys.argv[0]))
|
||||
app = Flask(__name__)
|
||||
app.config['storedir'] = storedir
|
||||
|
||||
|
@ -35,13 +36,14 @@ def init():
|
|||
@returns_plain
|
||||
@app.route("/", methods=["POST"])
|
||||
def store_file():
|
||||
try:
|
||||
#try:
|
||||
data = request.form['p']
|
||||
key = generate_key()
|
||||
store(key, data)
|
||||
return "http://%s/%s" % (request.host, key)
|
||||
except:
|
||||
return usage(request.host), 401
|
||||
return u"http://%s/%s\n" % (request.host, key)
|
||||
#except Exception as e:
|
||||
# print(e)
|
||||
# return usage(request.host), 401
|
||||
|
||||
|
||||
@returns_plain
|
||||
|
@ -55,19 +57,24 @@ def retrieve_file(key):
|
|||
|
||||
def store(key, value):
|
||||
with open("%s/%s" % (storedir, key), "wb+") as f:
|
||||
f.write(value)
|
||||
f.write(bytes(value, "UTF-8"))
|
||||
|
||||
|
||||
def retrieve(key):
|
||||
with open("%s/%s" % (storedir, key), "rb+") as f:
|
||||
return f.read()
|
||||
return f.read().decode("UTF-8")
|
||||
|
||||
|
||||
def generate_key():
|
||||
s = string.lowercase+string.digits
|
||||
return ''.join(random.sample(s, 24))
|
||||
s = string.ascii_lowercase+string.digits
|
||||
return ''.join(random.sample(s, 10))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
#app.debug = True
|
||||
print("Storedir is %s" % storedir)
|
||||
try:
|
||||
os.mkdir(storedir)
|
||||
except:
|
||||
pass
|
||||
app.run(host='0.0.0.0')
|
||||
|
|
Loading…
Reference in a new issue