irt/docker_nsrl
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Write changes to configfile

Browse Source
This commit is contained in:
TKE
2020-04-24 17:46:33 +02:00
parent 175dfce1f6
commit 41886a47af
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
nsrl/build.py
View File

@@ -38,7 +38,8 @@ def main():
args = parser.parse_args() args = parser.parse_args()
#check if config-file was given #check if config-file was given
configfiles=['/nsrl/nsrl.conf'] default_config_file='/nsrl/nsrl.conf'
configfiles=[default_config_file]
if not args.config is None: if not args.config is None:
#add user config #add user config
if os.path.isfile(args.config): if os.path.isfile(args.config):
@@ -77,6 +78,7 @@ def main():
_ = f_line.readline() _ = f_line.readline()
print "[BUILDING] Calculating number of entries in Inputfile..." print "[BUILDING] Calculating number of entries in Inputfile..."
num_lines = sum(bl.count("\n") for bl in blocks(f_line)) num_lines = sum(bl.count("\n") for bl in blocks(f_line))
conf['hash_count']=num_lines
else: else:
num_lines=conf.getint("hash_count") num_lines=conf.getint("hash_count")
print "[BUILDING] There are {} {}s in the Database".format(num_lines,hashfile_type) print "[BUILDING] There are {} {}s in the Database".format(num_lines,hashfile_type)
@@ -102,7 +104,9 @@ def main():
else: else:
print("[ERROR] No such file or directory: %s", nsrl_path) print("[ERROR] No such file or directory: %s", nsrl_path)
return #save config
with open(default_config_file,'w') as configfile:
config.write(configfile)
if __name__ == "__main__": if __name__ == "__main__":