Merge pull request #239 from CybercentreCanada/hotfix/pyroute2_import

Switch from pyroute2 to pyroute2.core

Author: cccs-jh

assemblyline/common/net.py

@@ -5,7 +5,7 @@
 from random import randint
 
 import netifaces as nif
-import pyroute2
+import pr2modules.iproute as iproute
 
 from assemblyline.common.net_static import TLDS_ALPHA_BY_DOMAIN
 
@@ -94,7 +94,7 @@ def get_random_mac(seperator: str = ':') -> str:
 def get_route_to(dst: str) -> str:
     ret_val = None
     try:
-        with pyroute2.IPRoute() as ipr:
+        with iproute.IPRoute() as ipr:
             for k, v in ipr.route('get', dst=dst)[0]['attrs']:
                 if k == "RTA_PREFSRC":
                     ret_val = v

setup.py

@@ -62,7 +62,7 @@
         'boto3',
         'pysftp',
         'netifaces',
-        'pyroute2',
+        'pyroute2.core',
         'redis',
         'requests',
         'elasticsearch>=7.0.0,<8.0.0,!=7.0.3',  # 7.0.3 is excluded due to an error