improves running list data

FeeiCN · Aug 11, 2017 · b2e2b90 · b2e2b90
1 parent 8a217ca
commit b2e2b90
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/cobra/__init__.py b/cobra/__init__.py
@@ -69,7 +69,7 @@ def main():
         if args.sid is None:
             a_sid = get_sid(args.target, True)
             data = {
-                'sids': []
+                'sids': {}
             }
             Running(a_sid).list(data)
             data = {

diff --git a/cobra/api.py b/cobra/api.py
@@ -94,7 +94,7 @@ def post():
             }
 
         a_sid_data = {
-            'sids': []
+            'sids': {}
         }
         running = Running(a_sid)
 
@@ -145,8 +145,8 @@ def post():
                 r_data = running.list()
                 ret = True
                 logger.info(r_data['sids'])
-                for s in r_data['sids']:
-                    if Running(s).is_file(True) is False:
+                for sid, git in r_data['sids'].items():
+                    if Running(sid).is_file(True) is False:
                         ret = False
                 if ret:
                     result['status'] = 'done'

diff --git a/cobra/cli.py b/cobra/cli.py
@@ -46,7 +46,7 @@ def start(target, formatter, output, special_rules, a_sid=None):
     s_sid = get_sid(target)
     r = Running(a_sid)
     data = r.list()
-    data['sids'].append(s_sid)
+    data['sids'][s_sid] = target
     r.list(data)
 
     report = '?sid={a_sid}'.format(a_sid=a_sid)