voussoir/etiquette
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add "Are you sure you want to quit" to erepl.

Browse source
This commit is contained in:
voussoir 2020-02-20 18:57:08 -08:00
parent 4a193d228c
commit ca8c642add
1 changed files with 13 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
frontends/etiquette_repl/etiquette_repl_launch.py
View file

@ -10,6 +10,8 @@ import os
import sys import sys
import traceback import traceback
from voussoirkit import getpermission
import etiquette import etiquette
P = etiquette.photodb.PhotoDB() P = etiquette.photodb.PhotoDB()
@ -38,9 +40,19 @@ get = P.get_tag
def erepl_argparse(args): def erepl_argparse(args):
if args.exec_statement: if args.exec_statement:
exec(args.exec_statement) exec(args.exec_statement)
P.commit()
else: else:
import code import code
while True:
try:
code.interact(banner='', local=dict(globals(), **locals())) code.interact(banner='', local=dict(globals(), **locals()))
except SystemExit:
pass
if len(P.savepoints) == 0:
break
print('You have uncommited changes, are you sure you want to quit?')
if getpermission.getpermission():
break
def main(argv): def main(argv):
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()