Cyberdefense/VulnWhisperer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Automatically create filepath and directory if it does not exist

Browse Source
This commit is contained in:
Austin Taylor
2018-02-25 21:15:50 -05:00
parent 5362d6f9e8
commit a0de072394
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
vulnwhisp/vulnwhisp.py
View File

@ -64,6 +64,14 @@ class vulnWhispererBase(object):
self.database = \ self.database = \
os.path.abspath(os.path.join(os.path.dirname(__file__), os.path.abspath(os.path.join(os.path.dirname(__file__),
'database', db_name)) 'database', db_name))
if not os.path.exists(self.db_path):
os.makedirs(self.db_path)
self.vprint('{info} Creating directory {dir}'.format(info=bcolors.INFO, dir=self.db_path))
if not os.path.exists(self.database):
with open(self.database, 'w'):
self.vprint('{info} Creating file {dir}'.format(info=bcolors.INFO, dir=self.database))
pass
try: try:
self.conn = sqlite3.connect(self.database) self.conn = sqlite3.connect(self.database)