Skip to content

Commit 661ffc5

Browse files
authored
Merge pull request #11 from pangerlkr/copilot/format-python-files-with-black
[WIP] Run Black formatter on Python files for code style conformity
2 parents c0fae8b + d4dd65e commit 661ffc5

File tree

4 files changed

+9
-27
lines changed

4 files changed

+9
-27
lines changed

gateway/app.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,7 @@
99
from pydantic import BaseModel
1010

1111
# Add modules to path
12-
sys.path.insert(
13-
0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "modules-python"))
14-
)
12+
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "modules-python")))
1513

1614
try:
1715
from ioc_analyzer import IOCAnalyzer

gateway/config.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,7 @@ class Settings(BaseSettings):
1919
redis_url: str = Field(default="redis://redis:6379/0", env="REDIS_URL")
2020

2121
# JWT
22-
jwt_secret: str = Field(
23-
default="change_me_in_production_minimum_32_characters", env="JWT_SECRET"
24-
)
22+
jwt_secret: str = Field(default="change_me_in_production_minimum_32_characters", env="JWT_SECRET")
2523
jwt_algorithm: str = Field(default="HS256", env="JWT_ALGORITHM")
2624
jwt_expiration_minutes: int = Field(default=60, env="JWT_EXPIRATION_MINUTES")
2725

@@ -31,9 +29,7 @@ class Settings(BaseSettings):
3129
api_workers: int = Field(default=4, env="API_WORKERS")
3230

3331
# CORS
34-
cors_origins: str = Field(
35-
default="http://localhost:3000,http://localhost:8000", env="CORS_ORIGINS"
36-
)
32+
cors_origins: str = Field(default="http://localhost:3000,http://localhost:8000", env="CORS_ORIGINS")
3733

3834
# Rate Limiting
3935
rate_limit_enabled: bool = Field(default=True, env="RATE_LIMIT_ENABLED")

modules-python/main.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,7 @@ def analyze_data():
3535
result = {"anomalies": anomalies, "total_analyzed": len(predictions)}
3636

3737
# Cache result
38-
redis_client.setex(
39-
f"analysis:{data.get('id', 'unknown')}", 3600, json.dumps(result)
40-
)
38+
redis_client.setex(f"analysis:{data.get('id', 'unknown')}", 3600, json.dumps(result))
4139

4240
return jsonify(result), 200
4341
else:

tests/test_python_modules.py

Lines changed: 5 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -41,16 +41,12 @@ def test_identify_md5_hash(self):
4141

4242
def test_identify_sha1_hash(self):
4343
"""Test SHA1 hash identification"""
44-
result = self.analyzer.identify_ioc_type(
45-
"da39a3ee5e6b4b0d3255bfef95601890afd80709"
46-
)
44+
result = self.analyzer.identify_ioc_type("da39a3ee5e6b4b0d3255bfef95601890afd80709")
4745
assert result == "sha1"
4846

4947
def test_identify_sha256_hash(self):
5048
"""Test SHA256 hash identification"""
51-
result = self.analyzer.identify_ioc_type(
52-
"e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855"
53-
)
49+
result = self.analyzer.identify_ioc_type("e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855")
5450
assert result == "sha256"
5551

5652
def test_identify_unknown(self):
@@ -133,12 +129,8 @@ def test_aggregate_threats(self):
133129
self.feed.add_feed("feed1", "https://example.com/feed1")
134130
self.feed.add_feed("feed2", "https://example.com/feed2")
135131

136-
self.feed.update_feed(
137-
"feed1", {"threats": [{"ip": "1.2.3.4", "severity": "high"}]}
138-
)
139-
self.feed.update_feed(
140-
"feed2", {"threats": [{"domain": "evil.com", "severity": "medium"}]}
141-
)
132+
self.feed.update_feed("feed1", {"threats": [{"ip": "1.2.3.4", "severity": "high"}]})
133+
self.feed.update_feed("feed2", {"threats": [{"domain": "evil.com", "severity": "medium"}]})
142134

143135
threats = self.feed.aggregate_threats()
144136
assert len(threats) == 2
@@ -168,9 +160,7 @@ def test_remove_nonexistent_feed(self):
168160
def test_fetch_indicators(self):
169161
"""Test fetching indicators"""
170162
self.feed.add_feed("test_feed", "https://example.com/feed")
171-
self.feed.update_feed(
172-
"test_feed", {"threats": [{"ip": "1.2.3.4", "severity": "high"}]}
173-
)
163+
self.feed.update_feed("test_feed", {"threats": [{"ip": "1.2.3.4", "severity": "high"}]})
174164
indicators = self.feed.fetch_indicators()
175165
assert len(indicators) == 1
176166
assert indicators[0]["ip"] == "1.2.3.4"

0 commit comments

Comments
 (0)