Add "Are you sure you want to quit" to erepl.
This commit is contained in:
		
							parent
							
								
									4a193d228c
								
							
						
					
					
						commit
						ca8c642add
					
				
					 1 changed files with 13 additions and 1 deletions
				
			
		|  | @ -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 | ||||||
|         code.interact(banner='', local=dict(globals(), **locals())) |         while True: | ||||||
|  |             try: | ||||||
|  |                 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() | ||||||
|  |  | ||||||
		Loading…
	
		Reference in a new issue