improves running list data

Author: FeeiCN

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 = {

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'

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)