mirror/0bin
1
0
Fork 0
mirror of https://github.com/Tygs/0bin.git synced 2023-08-10 21:13:00 +03:00
Code Issues Projects Releases Wiki Activity

Merge branch 'v2' of github.com:Tygs/0bin into v2

Browse Source
This commit is contained in:
papee
2020-08-12 09:34:14 +02:00
parent a8e6763cc0 722fb45b51
commit 2e67a09bca
134 changed files with 45 additions and 40037 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
zerobin/cmd.py
View File

@ -1,24 +1,15 @@
#!/usr/bin/env python
# coding: utf-8
#!/usr/bin/env python3
from __future__ import unicode_literals, absolute_import, print_function
"""
Main script including runserver and delete-paste.
"""
import sys
from sigtools.modifiers import annotate, autokwoargs
import re
import _thread as thread
try:
import thread
except ImportError:
import _thread as thread
from zerobin.utils import (settings, SettingsValidationError,
drop_privileges)
from zerobin.utils import settings, SettingsValidationError, drop_privileges
from zerobin.routes import get_app
from zerobin.paste import Paste
@ -26,13 +17,22 @@ from bottle import run
import clize
@annotate(debug=bool, compressed_static=bool)
def runserver(host='', port='', debug=None, user='', group='',
settings_file='', compressed_static=None,
version=False, paste_id_length=None, server="cherrypy"):
def runserver(
host="",
port="",
debug=None,
user="",
group="",
settings_file="",
compressed_static=None,
version=False,
paste_id_length=None,
server="cherrypy",
):
if version:
print('0bin V%s' % settings.VERSION)
print("0bin V%s" % settings.VERSION)
sys.exit(0)
settings.HOST = host or settings.HOST
@ -44,14 +44,19 @@ def runserver(host='', port='', debug=None, user='', group='',
try:
_, app = get_app(debug, settings_file, compressed_static, settings=settings)
except SettingsValidationError as err:
print('Configuration error: %s' % err.message, file=sys.stderr)
print("Configuration error: %s" % err.message, file=sys.stderr)
sys.exit(1)
thread.start_new_thread(drop_privileges, (settings.USER, settings.GROUP))
if settings.DEBUG:
run(app, host=settings.HOST, port=settings.PORT, reloader=True,
server="cherrypy")
run(
app,
host=settings.HOST,
port=settings.PORT,
reloader=True,
server="cherrypy",
)
else:
run(app, host=settings.HOST, port=settings.PORT, server="cherrypy")
@ -59,7 +64,8 @@ def runserver(host='', port='', debug=None, user='', group='',
# The regex parse the url and separate the paste's id from the decription key
# After the '/paste/' part, there is several caracters, identified as
# the uuid of the paste. Followed by a '#', the decryption key of the paste.
paste_url = re.compile('/paste/(?P<paste_id>.*)#(?P<key>.*)')
paste_url = re.compile("/paste/(?P<paste_id>.*)#(?P<key>.*)")
def unpack_paste(paste):
"""Take either the ID or the URL of a paste, and return its ID"""
@ -67,11 +73,10 @@ def unpack_paste(paste):
try_url = paste_url.search(paste)
if try_url:
return try_url.group('paste_id')
return try_url.group("paste_id")
return paste
@annotate(quiet='q')
@autokwoargs
def delete_paste(quiet=False, *pastes):
"""
Remove pastes, given its ID or its URL
@ -86,16 +91,19 @@ def delete_paste(quiet=False, *pastes):
Paste.load(paste_uuid).delete()
if not quiet:
print('Paste {} is removed'.format(paste_uuid))
print("Paste {} is removed".format(paste_uuid))
except ValueError:
if not quiet:
print('Paste {} doesn\'t exist'.format(paste_uuid))
print("Paste {} doesn't exist".format(paste_uuid))
def main():
subcommands = [runserver, delete_paste]
subcommand_names = [clize.util.name_py2cli(name)
for name in clize.util.dict_from_names(subcommands).keys()]
subcommand_names = [
clize.util.name_py2cli(name)
for name in clize.util.dict_from_names(subcommands).keys()
]
if len(sys.argv) < 2 or sys.argv[1] not in subcommand_names:
sys.argv.insert(1, subcommand_names[0])
clize.run(runserver, delete_paste)