Merge remote-tracking branch 'upstream/master'

main
Zak 2022-10-19 14:41:17 +02:00
commit 51b3958175
1 changed files with 14 additions and 0 deletions

View File

@ -328,7 +328,9 @@ class CMEDBMenu(cmd.Cmd):
pass
def do_workspace(self, line):
helpString = "[-] wordkspace create <targetName> | workspace list | workspace <targetName>"
if not line:
print(helpString)
return
line = line.strip()
@ -364,6 +366,14 @@ class CMEDBMenu(cmd.Cmd):
self.do_workspace(new_workspace)
elif line.split()[0] == 'list':
print("[*] Enumerating Workspaces")
for workspace in os.listdir(os.path.join(self.workspace_dir)):
if(workspace==self.workspace):
print("==> "+workspace)
else:
print(workspace)
elif os.path.exists(os.path.join(self.workspace_dir, line)):
self.config.set('CME', 'workspace', line)
self.write_configfile()
@ -371,6 +381,10 @@ class CMEDBMenu(cmd.Cmd):
self.workspace = line
self.prompt = 'cmedb ({}) > '.format(line)
else:
print(helpString)
def do_exit(self, line):
sys.exit(0)