specchioriflesso/bump
1
Fork 0
mirror of https://github.com/makefu/bump.git synced 2024-03-22 11:32:52 +01:00
Code Issues Projects Releases Packages Wiki Activity

fix store path

Browse source
This commit is contained in:
makefu 2013-11-19 09:07:18 +01:00
parent 4ba8428400
commit 8a0f30785a
1 changed files with 17 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
bump/init.py
View file

@ -1,11 +1,11 @@
from flask import Flask, Response, request from flask import Flask, Response, request, send_from_directory
from functools import wraps from functools import wraps
import sys import sys
import random import random
import string import string
import os import os
storedir = "%s/../store" % os.path.dirname(os.path.abspath(sys.argv[0]))
app = Flask(__name__) app = Flask(__name__)
storedir = os.path.join(app.root_path, 'store'),
app.config['storedir'] = storedir app.config['storedir'] = storedir
@ -18,12 +18,17 @@ def returns_plain(f):
def usage(host): def usage(host):
return """Usage: return """Bump Usage:
> echo balls | curl -F "p=<-" {0} > echo balls | curl -F "p=<-" {0}
< http://{0}/q9a1bfljzm < http://{0}/q9a1bfljzm
> curl {0}/q9a1bfljzm > curl {0}/q9a1bfljzm
< balls""".format(host) < balls
Source at http://github.com/makefu/bump""".format(host)
@app.route('/favicon.ico')
def favicon():
return send_from_directory(os.path.join(app.root_path, 'static'),
'favicon.ico', mimetype='image/vnd.microsoft.icon')
@returns_plain @returns_plain
@app.route("/", methods=["GET"]) @app.route("/", methods=["GET"])
@ -40,7 +45,7 @@ def store_file():
data = request.form['p'] data = request.form['p']
key = generate_key() key = generate_key()
store(key, data) store(key, data)
return u"http://%s/%s\n" % (request.host, key) return "http://%s/%s\n" % (request.host, key)
#except Exception as e: #except Exception as e:
# print(e) # print(e)
# return usage(request.host), 401 # return usage(request.host), 401
@ -57,7 +62,12 @@ def retrieve_file(key):
def store(key, value): def store(key, value):
with open("%s/%s" % (storedir, key), "wb+") as f: with open("%s/%s" % (storedir, key), "wb+") as f:
f.write(bytes(value, "UTF-8")) try:
# py3
f.write(bytes(value, "UTF-8"))
except:
# py2
f.write(value)
def retrieve(key): def retrieve(key):
@ -71,7 +81,7 @@ def generate_key():
if __name__ == "__main__": if __name__ == "__main__":
#app.debug = True app.debug = True
print("Storedir is %s" % storedir) print("Storedir is %s" % storedir)
try: try:
os.mkdir(storedir) os.mkdir(storedir)