diff --git a/app/templates/project.html b/app/templates/project.html index 1e67c05..0d7710f 100644 --- a/app/templates/project.html +++ b/app/templates/project.html @@ -12,14 +12,18 @@
- +
- + +
+ + + +
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)