Fix one more old style import

This commit is contained in:
Daniel Goldberg 2018-08-27 11:06:58 -04:00
parent b23418782c
commit cad9aca5dd
1 changed files with 2 additions and 1 deletions

View File

@ -3,6 +3,7 @@ import socket
from infection_monkey.model.host import VictimHost from infection_monkey.model.host import VictimHost
from infection_monkey.network import HostFinger from infection_monkey.network import HostFinger
import infection_monkey.config
__author__ = 'Maor Rayzin' __author__ = 'Maor Rayzin'
@ -18,7 +19,7 @@ class MSSQLFinger(HostFinger):
SERVICE_NAME = 'MSSQL' SERVICE_NAME = 'MSSQL'
def __init__(self): def __init__(self):
self._config = __import__('config').WormConfiguration self._config = infection_monkey.config.WormConfiguration
def get_host_fingerprint(self, host): def get_host_fingerprint(self, host):
"""Gets Microsoft SQL Server instance information by querying the SQL Browser service. """Gets Microsoft SQL Server instance information by querying the SQL Browser service.