diff --git a/app/views.py b/app/views.py
index 112ece6..49d0aea 100644
--- a/app/views.py
+++ b/app/views.py
@@ -63,7 +63,6 @@ def project(name):
@views.route("/add_project", methods=['POST', 'GET'])
def inject():
-
if request.method == 'POST':
config.load()
settings = Settings()
@@ -88,15 +87,19 @@ def inject():
inject_style = request.form['inject_style']
settings.inject_style = InjectStyle[inject_style]
-
- print(str(settings))
-
- project = Project(project_name, settings)
- project.settings = settings
- storage.add_project(project)
+
+ if storage.get_project(project_name) != None:
+ project = storage.get_project(project_name)
+ project.settings = settings
+ else:
+ project = Project(project_name, settings)
+ project.settings = settings
+ settings.project_name = project_name
+ storage.add_project(project)
storage.save_data()
- return render_template('project_add_post.html')
- else:
+ return redirect("/project/{}".format(project_name), code=302)
+
+ else: # GET
exes = []
for file in os.listdir("app/upload/exe"):
exes.append(file)
@@ -122,7 +125,6 @@ def inject():
)
#start(settings)
-
@views.route("/build")
diff --git a/model/project.py b/model/project.py
index 961ed22..dd480b6 100644
--- a/model/project.py
+++ b/model/project.py
@@ -11,6 +11,7 @@ logger = logging.getLogger("Project")
class Project():
def __init__(self, settings: Settings):
+ self.name: str = ""
self.settings: Settings = settings
self.payload: Payload = Payload(self.settings.payload_path)
self.exe_host: ExeHost = ExeHost(self.settings.inject_exe_in)