Skip to content

Commit c57217d

Browse files
committed
remove update script. refactored loading mechanism
1 parent 617b967 commit c57217d

File tree

3 files changed

+16
-106
lines changed

3 files changed

+16
-106
lines changed

cirtkit.py

+14-4
Original file line numberDiff line numberDiff line change
@@ -1,20 +1,30 @@
11
#!/usr/bin/env python
2-
# This file is part of Viper - https://github.com/cirtkit-framework/cirtkit
3-
# See the file 'LICENSE' for copying permission.
2+
# This file is part of Cirtkit - https://github.com/cirtkit-framework/cirtkit
43

54
import argparse
5+
from sys import argv, executable
6+
from os import path, symlink, getcwd
67

78
from lib.core.ui import console
89
from lib.core.investigation import __project__
910
from lib.core.database import Database
1011

12+
if argv[0][-3:] == '.py':
13+
# create symlink
14+
try:
15+
src = getcwd() + '/' + argv[0]
16+
dst = executable
17+
print dst
18+
except:
19+
pass
20+
1121
parser = argparse.ArgumentParser()
12-
parser.add_argument('-i', '--investigation', help='Specify a new or existing investigation name', action='store', required=False)
22+
parser.add_argument('-i', '--investigation', help='Specify a new or existing investigation', action='store', required=False)
1323
args = parser.parse_args()
1424

1525
if args.investigation:
1626
db = Database()
1727
__project__.open(args.investigation, db)
1828

1929
c = console.Console()
20-
c.start()
30+
c.start()

lib/core/plugins.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ def load_modules():
1313
import modules
1414

1515
moduleDict = dict()
16+
integrationDict = dict()
1617

1718
# Walk recursively through all modules and packages.
1819
for loader, module_name, ispkg in pkgutil.walk_packages(modules.__path__, modules.__name__ + '.'):
@@ -34,7 +35,7 @@ def load_modules():
3435
# Yield the class if it's a subclass of Module.
3536
if issubclass(member_object, Module) and member_object is not Module:
3637
moduleDict[member_object.cmd] = dict(obj=member_object, description=member_object.description)
37-
38+
3839
return moduleDict
3940

4041

update.py

-101
This file was deleted.

0 commit comments

Comments
 (0)