Skip to content

Commit 0ba6d98

Browse files
improve naming between frontend/controller
1 parent 6ebce95 commit 0ba6d98

File tree

7 files changed

+89
-171
lines changed

7 files changed

+89
-171
lines changed

nebula/addons/attacks/attacks.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -129,8 +129,10 @@ def create_attack(engine) -> Attack:
129129
}
130130

131131
# Get attack name and parameters from the engine configuration
132-
attack_name = engine.config.participant["adversarial_args"]["attacks"]
133-
attack_params = engine.config.participant["adversarial_args"].get("attack_params", {}).items()
132+
attack_params = engine.config.participant["adversarial_args"].get("attack_params", {})
133+
attack_name = attack_params.get("attacks", None)
134+
if attack_name is None:
135+
raise AttackException("No attack specified")
134136

135137
# Look up the attack class based on the attack name
136138
attack = ATTACK_MAP.get(attack_name)

nebula/controller/controller.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,6 @@
2525
from nebula.config.config import Config
2626
from nebula.config.mender import Mender
2727
from nebula.controller.database import scenario_set_all_status_to_finished, scenario_set_status_to_finished
28-
from nebula.controller.scenarios import Scenario, ScenarioManagement
2928
from nebula.utils import DockerUtils, FileUtils, SocketUtils
3029

3130

@@ -313,6 +312,7 @@ async def remove_scenario(
313312
dict: A message indicating successful removal.
314313
"""
315314
from nebula.controller.database import remove_scenario_by_name
315+
from nebula.controller.scenarios import ScenarioManagement
316316

317317
try:
318318
remove_scenario_by_name(scenario_name)
@@ -396,6 +396,7 @@ async def update_scenario(
396396
dict: A message confirming the update.
397397
"""
398398
from nebula.controller.database import scenario_update_record
399+
from nebula.controller.scenarios import Scenario
399400

400401
try:
401402
scenario = Scenario.from_dict(scenario)
@@ -1631,6 +1632,8 @@ def stop():
16311632
- Handles any exceptions during PID reading or killing by logging them.
16321633
- Exits the program with status code 0.
16331634
"""
1635+
from nebula.controller.scenarios import ScenarioManagement
1636+
16341637
logging.info("Closing NEBULA (exiting from components)... Please wait")
16351638
DockerUtils.remove_containers_by_prefix(f"{os.environ['USER']}_")
16361639
ScenarioManagement.stop_blockchain()

nebula/controller/database.py

Lines changed: 1 addition & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -141,10 +141,6 @@ async def initialize_databases(databases_dir):
141141
network_subnet TEXT,
142142
network_gateway TEXT,
143143
epochs TEXT,
144-
attacks TEXT,
145-
poisoned_node_percent TEXT,
146-
poisoned_sample_percent TEXT,
147-
poisoned_noise_percent TEXT,
148144
attack_params TEXT,
149145
with_reputation TEXT,
150146
random_geo TEXT,
@@ -193,10 +189,6 @@ async def initialize_databases(databases_dir):
193189
"network_subnet": "TEXT",
194190
"network_gateway": "TEXT",
195191
"epochs": "TEXT",
196-
"attacks": "TEXT",
197-
"poisoned_node_percent": "TEXT",
198-
"poisoned_sample_percent": "TEXT",
199-
"poisoned_noise_percent": "TEXT",
200192
"attack_params": "TEXT",
201193
"with_reputation": "TEXT",
202194
"random_geo": "TEXT",
@@ -592,10 +584,6 @@ def scenario_update_record(
592584
network_subnet,
593585
network_gateway,
594586
epochs,
595-
attacks,
596-
poisoned_node_percent,
597-
poisoned_sample_percent,
598-
poisoned_noise_percent,
599587
attack_params,
600588
with_reputation,
601589
random_geo,
@@ -613,7 +601,7 @@ def scenario_update_record(
613601
role,
614602
username
615603
) VALUES (
616-
?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?
604+
?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?
617605
);
618606
"""
619607
c.execute(insert_command, (
@@ -644,10 +632,6 @@ def scenario_update_record(
644632
scenario.network_subnet,
645633
scenario.network_gateway,
646634
scenario.epochs,
647-
json.dumps(scenario.attacks),
648-
scenario.poisoned_node_percent,
649-
scenario.poisoned_sample_percent,
650-
scenario.poisoned_noise_percent,
651635
json.dumps(scenario.attack_params),
652636
scenario.with_reputation,
653637
scenario.random_geo,
@@ -694,10 +678,6 @@ def scenario_update_record(
694678
network_subnet = ?,
695679
network_gateway = ?,
696680
epochs = ?,
697-
attacks = ?,
698-
poisoned_node_percent = ?,
699-
poisoned_sample_percent = ?,
700-
poisoned_noise_percent = ?,
701681
attack_params = ?,
702682
with_reputation = ?,
703683
random_geo = ?,
@@ -743,7 +723,6 @@ def scenario_update_record(
743723
scenario.network_subnet,
744724
scenario.network_gateway,
745725
scenario.epochs,
746-
json.dumps(scenario.attacks),
747726
scenario.poisoned_node_percent,
748727
scenario.poisoned_sample_percent,
749728
scenario.poisoned_noise_percent,

0 commit comments

Comments
 (0)