91 lines
1.8 KiB
Python
91 lines
1.8 KiB
Python
#!/usr/bin/python3.6
|
|
|
|
from flask import Flask, request, render_template
|
|
import random, string
|
|
from pathlib import Path
|
|
from pprint import pprint
|
|
|
|
# Basic settings for paste
|
|
|
|
pasteDir = Path("pastes")
|
|
nameLength = 10
|
|
useSpecialChars = False
|
|
allowListing = False
|
|
hostedURL = "http://paste.benjamyn.love/"
|
|
|
|
letters = string.ascii_letters
|
|
if useSpecialChars:
|
|
letters = letters + string.punctuation
|
|
|
|
# Functions
|
|
def randomName():
|
|
name = ""
|
|
for x in range(0,nameLength):
|
|
num = random.randint(0,len(letters))
|
|
print(num)
|
|
name += letters[num - 1]
|
|
return name
|
|
|
|
def doesDirExist():
|
|
if pasteDir.is_dir() == False:
|
|
pasteDir.mkdir()
|
|
|
|
def doesPasteExist(pasteName):
|
|
pasten = pasteDir / pasteName
|
|
if pasten.is_file():
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def writePaste(pasteName, pasteData):
|
|
pasten = pasteDir / pasteName
|
|
tmpData = pasteData.split('\n')
|
|
for line in tmpData:
|
|
with pasten.open('a') as f: f.write(line + '\n')
|
|
|
|
|
|
|
|
doesDirExist()
|
|
|
|
# Web interface
|
|
|
|
# https://paste.benjamyn.love/CBynvvjwDK
|
|
|
|
app = Flask(__name__)
|
|
app.config['DEBUG'] = False
|
|
|
|
@app.route('/', methods=['POST', 'GET'])
|
|
def index():
|
|
if request.method == "GET":
|
|
if allowListing == True:
|
|
return "FileList"
|
|
else:
|
|
return render_template("post.html")
|
|
elif request.method == "POST":
|
|
data = request.get_data().decode()
|
|
if "buttonwaspushed" in data:
|
|
data = request.form['textarea']
|
|
name = randomName()
|
|
while doesPasteExist(name):
|
|
name = randomName()
|
|
else:
|
|
print("Writing: " + data)
|
|
writePaste(name, data)
|
|
return hostedURL + name
|
|
|
|
|
|
@app.route('/<path:path>')
|
|
def getPaste(path):
|
|
tmpPath = Path(path)
|
|
pasten = pasteDir / tmpPath.name
|
|
print("Checking " + str(pasten))
|
|
if pasten.is_file():
|
|
with pasten.open() as f: data = f.read()
|
|
print(data)
|
|
return render_template("read.html", data=data)
|
|
else:
|
|
return "No paste found"
|
|
|
|
if __name__ == '__main__':
|
|
app.run(host='127.0.0.1')
|