lint

Author: pinheadmz

src/warnet/admin.py

@@ -17,7 +17,6 @@
 )
 from .namespaces import copy_namespaces_defaults, namespaces
 from .network import copy_network_defaults
-from .process import run_command
 
 
 @click.group(name="admin", hidden=True)
@@ -114,7 +113,11 @@ def create_kubeconfigs(kubeconfig_dir, token_duration):
                 "contexts": [
                     {
                         "name": f"{name}-{namespace}",
-                        "context": {"cluster": cluster["name"], "namespace": namespace, "user": name},
+                        "context": {
+                            "cluster": cluster["name"],
+                            "namespace": namespace,
+                            "user": name,
+                        },
                     }
                 ],
                 "current-context": f"{name}-{namespace}",

src/warnet/k8s.py

@@ -4,7 +4,7 @@
 import tarfile
 import tempfile
 from pathlib import Path
-from time import time, sleep
+from time import sleep, time
 from typing import Optional
 
 import yaml
@@ -524,18 +524,22 @@ def get_warnet_user_service_accounts_in_namespace(namespace):
     """
     sclient = get_static_client()
     sas = sclient.list_namespaced_service_account(namespace)
-    return [sa for sa in sas.items if sa.metadata.labels and "mission" in sa.metadata.labels and sa.metadata.labels["mission"] == "user"]
+    return [
+        sa
+        for sa in sas.items
+        if sa.metadata.labels
+        and "mission" in sa.metadata.labels
+        and sa.metadata.labels["mission"] == "user"
+    ]
+
 
 def get_token_for_service_acount(sa, duration):
     sclient = get_static_client()
     spec = V1TokenRequestSpec(
-        audiences=["https://kubernetes.default.svc"],
-        expiration_seconds=duration
+        audiences=["https://kubernetes.default.svc"], expiration_seconds=duration
     )
     resp = sclient.create_namespaced_service_account_token(
-        name=sa.metadata.name,
-        namespace=sa.metadata.namespace,
-        body=spec
+        name=sa.metadata.name, namespace=sa.metadata.namespace, body=spec
     )
     return resp.status.token
 

test/wargames_test.py

@@ -98,6 +98,7 @@ def check_scenario_permissions(self):
                 count += 1
                 assert count <= 1
 
+
 if __name__ == "__main__":
     test = WargamesTest()
     test.run_test()