correct write

CIRCL · May 20, 2021 · 28d4fd1 · 28d4fd1
1 parent 10b2c96
commit 28d4fd1
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 9 deletions.
diff --git a/bin/Generator.py b/bin/Generator.py
@@ -6,9 +6,9 @@
 import datetime
 import subprocess
 import pathlib
-p = pathlib.Path(__file__).parent.absolute()
+pathProg = pathlib.Path(__file__).parent.absolute()
 s = ""
-for i in re.split(r"/|\\", str(p))[:-1]:
+for i in re.split(r"/|\\", str(pathProg))[:-1]:
     s += i + "/"
 sys.path.append(s + "etc")
 import allVariables
@@ -23,7 +23,7 @@ def runningVms():
     return subprocess.run(req, capture_output=True)
 
 def readFile():
-    f = open(os.path.dirname(sys.argv[0]) + "/tmp","r")
+    f = open(pathProg + "/tmp","r")
     l = f.readline().rstrip()
     f.close()
     return l

diff --git a/bin/automatisation_yara.py b/bin/automatisation_yara.py
@@ -1,4 +1,3 @@
-import os
 import re
 import string
 import datetime

diff --git a/bin/get_pe.py b/bin/get_pe.py
@@ -9,8 +9,6 @@
     s += i + "/"
 sys.path.append(s + "etc")
 import allVariables
-sys.path.append(s + "etc")
-import allVariables
 
 def pe_yara(file_pe):
     pe =  pefile.PE(file_pe)

diff --git a/bin/serveur.py b/bin/serveur.py
@@ -4,9 +4,9 @@
 import time
 import flask
 import pathlib
-p = pathlib.Path(__file__).parent.absolute()
+pathProg = pathlib.Path(__file__).parent.absolute()
 s = ""
-for i in re.split(r"/|\\", str(p))[:-1]:
+for i in re.split(r"/|\\", str(pathProg))[:-1]:
     s += i + "/"
 sys.path.append(s + "etc")
 import allVariables
@@ -32,7 +32,7 @@
 #exit(0)
 
 def WriteFileP(s, current):
-    f = open(os.path.dirname(sys.argv[0]) + "/tmp", "w")
+    f = open(pathProg + "/tmp", "w")
     f.write(s + ":" + current)
     f.close()