isort

maybe-hello-world · maybe-hello-world · commit a925eaa05692 · 2023-11-25T23:11:21.000-08:00
diff --git a/tasks/capture/tcpdump.py b/tasks/capture/tcpdump.py
@@ -1,16 +1,16 @@
+import signal
 import subprocess
 import time
-import signal
 from typing import List, Optional
 
 from netunicorn.base import (
-    Task,
-    TaskDispatcher,
-    Result,
-    Success,
+    Architecture,
     Failure,
     Node,
-    Architecture,
+    Result,
+    Success,
+    Task,
+    TaskDispatcher,
 )
 from netunicorn.library.tasks.tasks_utils import subprocess_run
 
diff --git a/tasks/capture/tshark.py b/tasks/capture/tshark.py
@@ -3,13 +3,13 @@
 from typing import List, Optional
 
 from netunicorn.base import (
-    Task,
-    TaskDispatcher,
-    Result,
-    Success,
+    Architecture,
     Failure,
     Node,
-    Architecture,
+    Result,
+    Success,
+    Task,
+    TaskDispatcher,
 )
 from netunicorn.library.tasks.tasks_utils import subprocess_run
 
diff --git a/tasks/flags.py b/tasks/flags.py
@@ -1,5 +1,5 @@
-from typing import Literal, Optional
 import warnings
+from typing import Literal, Optional
 from urllib.parse import quote_plus
 
 from netunicorn.base import Task
@@ -27,9 +27,10 @@ def __init__(self, flag_name: str, flag_values: FlagValues, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
     def run(self) -> None:
-        import requests as req
         import os
 
+        import requests as req
+
         gateway = os.environ["NETUNICORN_GATEWAY_ENDPOINT"]
         experiment_id = os.environ["NETUNICORN_EXPERIMENT_ID"]
 
@@ -45,9 +46,10 @@ def __init__(self, flag_name: str, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
     def run(self) -> FlagValues:
-        import requests as req
         import os
 
+        import requests as req
+
         gateway = os.environ["NETUNICORN_GATEWAY_ENDPOINT"]
         experiment_id = os.environ["NETUNICORN_EXPERIMENT_ID"]
 
@@ -75,10 +77,11 @@ def __init__(
         super().__init__(*args, **kwargs)
 
     def run(self) -> FlagValues:
-        import requests as req
         import os
         import time
 
+        import requests as req
+
         gateway = os.environ["NETUNICORN_GATEWAY_ENDPOINT"]
         experiment_id = os.environ["NETUNICORN_EXPERIMENT_ID"]
 
@@ -118,9 +121,10 @@ def __init__(
         super().__init__(*args, **kwargs)
 
     def run(self) -> None:
-        import requests as req
         import os
 
+        import requests as req
+
         gateway = os.environ["NETUNICORN_GATEWAY_ENDPOINT"]
         experiment_id = os.environ["NETUNICORN_EXPERIMENT_ID"]
 
diff --git a/tasks/letsencrypt/__init__.py b/tasks/letsencrypt/__init__.py
@@ -1,4 +1,5 @@
 from netunicorn.base.task import Task
+
 from .tasks import validate_dns_01, validate_http_01
 
 
diff --git a/tasks/letsencrypt/tasks.py b/tasks/letsencrypt/tasks.py
@@ -1,5 +1,5 @@
 import requests
-from returns.result import Success, Failure, Result
+from returns.result import Failure, Result, Success
 
 
 def validate_http_01(domain: str, token_name: str, token_data: str) -> Result[None, str]:
diff --git a/tasks/measurements/cloudflare/speedtest.py b/tasks/measurements/cloudflare/speedtest.py
@@ -5,7 +5,7 @@
 from dataclasses import dataclass
 from typing import Optional
 
-from netunicorn.base import Result, Task, TaskDispatcher, Node, Architecture, Success
+from netunicorn.base import Architecture, Node, Result, Success, Task, TaskDispatcher
 
 
 @dataclass
diff --git a/tasks/measurements/iperf3.py b/tasks/measurements/iperf3.py
@@ -2,7 +2,7 @@
 import time
 from typing import Optional
 
-from netunicorn.base import Task, Success, Failure
+from netunicorn.base import Failure, Success, Task
 from netunicorn.library.tasks.tasks_utils import subprocess_run
 
 
diff --git a/tasks/measurements/ndt.py b/tasks/measurements/ndt.py
@@ -1,6 +1,6 @@
 from typing import Optional
 
-from netunicorn.base import Node, Architecture, Task, TaskDispatcher
+from netunicorn.base import Architecture, Node, Task, TaskDispatcher
 from netunicorn.library.tasks.tasks_utils import subprocess_run
 
 __all__ = [
diff --git a/tasks/network_attacks/arp/__init__.py b/tasks/network_attacks/arp/__init__.py
@@ -1,4 +1,5 @@
 from netunicorn.base.task import Task
+
 from .spoof import main
 
 
diff --git a/tasks/network_attacks/arp/spoof.py b/tasks/network_attacks/arp/spoof.py
@@ -1,7 +1,7 @@
 import time
 
 from scapy.all import sendp
-from scapy.layers.l2 import Ether, ARP
+from scapy.layers.l2 import ARP, Ether
 
 
 def main(
diff --git a/tasks/network_attacks/heartbleed/__init__.py b/tasks/network_attacks/heartbleed/__init__.py
@@ -1,7 +1,9 @@
 import time
 from enum import Enum
-from netunicorn.base.task import Task, Success, Failure
-from .heartbleeder import connect, send_hello, bleed
+
+from netunicorn.base.task import Failure, Success, Task
+
+from .heartbleeder import bleed, connect, send_hello
 
 
 class TLSVersion(Enum):
diff --git a/tasks/network_attacks/heartbleed/heartbleeder.py b/tasks/network_attacks/heartbleed/heartbleeder.py
@@ -13,12 +13,12 @@
 
 import codecs
 import random
-import sys
-import struct
-import socket
 import re
+import socket
+import struct
+import sys
 import time
-from typing import Iterable, List, Tuple, Optional
+from typing import Iterable, List, Optional, Tuple
 
 #
 decode_hex = codecs.getdecoder("hex_codec")
diff --git a/tasks/network_attacks/icmp/__init__.py b/tasks/network_attacks/icmp/__init__.py
@@ -1,4 +1,5 @@
 from netunicorn.base.task import Task
+
 from .redirection import main
 
 
diff --git a/tasks/network_attacks/icmp/redirection.py b/tasks/network_attacks/icmp/redirection.py
@@ -1,4 +1,4 @@
-from scapy.layers.inet import IP, ICMP
+from scapy.layers.inet import ICMP, IP
 from scapy.sendrecv import send
 
 
diff --git a/tasks/network_attacks/land/__init__.py b/tasks/network_attacks/land/__init__.py
@@ -1,4 +1,5 @@
 from netunicorn.base.task import Task
+
 from .landattack import land_attack
 
 
diff --git a/tasks/network_attacks/loris/__init__.py b/tasks/network_attacks/loris/__init__.py
@@ -1,4 +1,5 @@
 from netunicorn.base import Task
+
 from .slowloris import main
 from .smbloris import smbloris
 
diff --git a/tasks/network_attacks/loris/slowloris.py b/tasks/network_attacks/loris/slowloris.py
@@ -5,8 +5,8 @@
 import logging
 import random
 import socket
-import time
 import ssl
+import time
 
 
 def send_line(self, line):
diff --git a/tasks/network_attacks/loris/smbloris.py b/tasks/network_attacks/loris/smbloris.py
@@ -1,5 +1,4 @@
 from scapy.all import *
-
 from scapy.layers.inet import IP, TCP
 
 
diff --git a/tasks/network_attacks/mac/__init__.py b/tasks/network_attacks/mac/__init__.py
@@ -1,4 +1,5 @@
 from netunicorn.base.task import Task
+
 from .flooder import flood
 
 
diff --git a/tasks/network_attacks/mac/flooder.py b/tasks/network_attacks/mac/flooder.py
@@ -1,7 +1,7 @@
-from scapy.layers.inet import IP, ICMP
+from scapy.layers.inet import ICMP, IP
 from scapy.layers.l2 import Ether
 from scapy.sendrecv import sendp
-from scapy.volatile import RandMAC, RandIP
+from scapy.volatile import RandIP, RandMAC
 
 
 def flood(iface: str = "eth0", count: int = 1000):
diff --git a/tasks/network_attacks/mail/__init__.py b/tasks/network_attacks/mail/__init__.py
@@ -1,7 +1,7 @@
-from .fake_mail import send_mail
-
 from netunicorn.base.task import Task
 
+from .fake_mail import send_mail
+
 
 class FakeMail(Task):
     def __init__(
diff --git a/tasks/preprocessing/scapy.py b/tasks/preprocessing/scapy.py
@@ -18,7 +18,7 @@ def __init__(self, filename: str, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
     def run(self) -> list[tuple[str, str, int, int, str]]:
-        from scapy.all import rdpcap, TCP, UDP, IP
+        from scapy.all import IP, TCP, UDP, rdpcap
 
         packets = rdpcap(self.filename)
         tuples = []
@@ -49,7 +49,7 @@ def __init__(self, filename: str, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
     def run(self) -> list[str]:
-        from scapy.all import rdpcap, DNSQR
+        from scapy.all import DNSQR, rdpcap
 
         pcap = rdpcap(self.filename)
         return [pkt[DNSQR].qname.decode() for pkt in pcap if DNSQR in pkt]
@@ -61,7 +61,7 @@ def __init__(self, filename: str, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
     def run(self) -> list[bytes]:
-        from scapy.all import rdpcap, Raw
+        from scapy.all import Raw, rdpcap
 
         packets = rdpcap(self.filename)
         return [
@@ -77,7 +77,7 @@ def __init__(self, filename: str, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
     def run(self) -> list[bytes]:
-        from scapy.all import rdpcap, ICMP
+        from scapy.all import ICMP, rdpcap
 
         packets = rdpcap(self.filename)
         return [pkt for pkt in packets if ICMP in pkt and pkt[ICMP].type == 8]
@@ -89,7 +89,7 @@ def __init__(self, filename: str, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
     def run(self) -> set:
-        from scapy.all import rdpcap, ARP
+        from scapy.all import ARP, rdpcap
 
         packets = rdpcap(self.filename)
         return {pkt[ARP].hwsrc for pkt in packets if ARP in pkt}
diff --git a/tasks/preprocessing/tshark.py b/tasks/preprocessing/tshark.py
@@ -1,5 +1,5 @@
+from netunicorn.base import Result, Task
 from netunicorn.library.tasks.tasks_utils import subprocess_run
-from netunicorn.base import Task, Result
 
 
 class TsharkCommand(Task):
diff --git a/tasks/preprocessing/zeek.py b/tasks/preprocessing/zeek.py
@@ -1,7 +1,7 @@
 from abc import ABC
 from typing import Optional
 
-from netunicorn.base import Task, TaskDispatcher, Result, Node, Architecture
+from netunicorn.base import Architecture, Node, Result, Task, TaskDispatcher
 from netunicorn.library.tasks.tasks_utils import subprocess_run
 
 
diff --git a/tasks/qoe_youtube/__init__.py b/tasks/qoe_youtube/__init__.py
@@ -4,9 +4,9 @@
 from typing import Optional
 
 from jinja2 import Environment, FileSystemLoader
+from netunicorn.base import Failure, Success, Task, TaskDispatcher, is_successful
 from netunicorn.base.architecture import Architecture
 from netunicorn.base.nodes import Node
-from netunicorn.base import Failure, Success, Task, TaskDispatcher, is_successful
 
 
 class StartQoECollectionServer(TaskDispatcher):
diff --git a/tasks/tasks_utils.py b/tasks/tasks_utils.py
@@ -1,5 +1,6 @@
 import subprocess
-from netunicorn.base import Result, Success, Failure
+
+from netunicorn.base import Failure, Result, Success
 
 
 def subprocess_run(arguments: list[str]) -> Result:
diff --git a/tasks/upload/webdav.py b/tasks/upload/webdav.py
@@ -1,7 +1,7 @@
 import os
 from typing import Literal, Optional, Set
 
-from netunicorn.base import Architecture, Node, Task, TaskDispatcher, Success, Failure
+from netunicorn.base import Architecture, Failure, Node, Success, Task, TaskDispatcher
 from netunicorn.library.tasks.tasks_utils import subprocess_run
 
 
diff --git a/tasks/utils/sleep.py b/tasks/utils/sleep.py
@@ -1,4 +1,5 @@
 import random
+
 from netunicorn.base import Task, TaskDispatcher
 from netunicorn.base.nodes import Node
 from netunicorn.library.tasks.basic import SleepTask
diff --git a/tasks/video_watchers/__init__.py b/tasks/video_watchers/__init__.py
@@ -1,3 +1,3 @@
-from .youtube_watcher import WatchYouTubeVideo, WatchYouTubeVideoLinuxImplementation
-from .vimeo_watcher import WatchVimeoVideo, WatchVimeoVideoLinuxImplementation
 from .twitch_watcher import WatchTwitchStream, WatchTwitchStreamLinuxImplementation
+from .vimeo_watcher import WatchVimeoVideo, WatchVimeoVideoLinuxImplementation
+from .youtube_watcher import WatchYouTubeVideo, WatchYouTubeVideoLinuxImplementation
diff --git a/tasks/video_watchers/vimeo_watcher.py b/tasks/video_watchers/vimeo_watcher.py
@@ -7,7 +7,7 @@
 import time
 from typing import Optional
 
-from netunicorn.base import Result, Failure, Success, Task, TaskDispatcher
+from netunicorn.base import Failure, Result, Success, Task, TaskDispatcher
 from netunicorn.base.architecture import Architecture
 from netunicorn.base.nodes import Node
 
diff --git a/tasks/video_watchers/youtube_watcher.py b/tasks/video_watchers/youtube_watcher.py
@@ -5,10 +5,10 @@
 import random
 import subprocess
 import time
-from typing import Optional
 from enum import IntEnum
+from typing import Optional
 
-from netunicorn.base import Result, Failure, Success, Task, TaskDispatcher
+from netunicorn.base import Failure, Result, Success, Task, TaskDispatcher
 from netunicorn.base.architecture import Architecture
 from netunicorn.base.nodes import Node
 

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`from netunicorn.base.task import Task`
	`2`	`+`
`2`	`3`	`from .tasks import validate_dns_01, validate_http_01`
`3`	`4`
`4`	`5`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`from netunicorn.base.task import Task`
	`2`	`+`
`2`	`3`	`from .spoof import main`
`3`	`4`
`4`	`5`