From 7509506f0b8d771bf0fe4875356e9c23acda3e1a Mon Sep 17 00:00:00 2001 From: 尹姜谊 Date: Wed, 17 Jan 2024 14:49:14 +0800 Subject: Add:增加geckovpn_serverip插件 MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- detection/vpn_detector.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) (limited to 'detection/vpn_detector.py') diff --git a/detection/vpn_detector.py b/detection/vpn_detector.py index 34cd307..6f76810 100644 --- a/detection/vpn_detector.py +++ b/detection/vpn_detector.py @@ -83,13 +83,14 @@ class VpnDetector: result_df['is_valid'] = 1 # result save - if len(result_df) > 1: + if len(result_df) > 0: self.logger.info('Start to update data to knowledgebase') result_path = os.path.join('data', plugin_name) if not os.path.exists(result_path): os.makedirs(result_path) result_file = os.path.join(result_path, output_filename) result_df.to_csv(result_file, index=False) + self.logger.info('Write {} {} {} to local file {} successfully.'.format(len(result_df), vpn_service_name, object_type, result_file)) # update to knowledgebase knowledge_api = KnowledgeApi(self.config) @@ -190,6 +191,15 @@ if __name__ == '__main__': elif plugin_name == 'ipvanishvpn_serverip': from vpnservices.ipvanishvpn_serverip import IpvanishvpnServerip detector = IpvanishvpnServerip(start_time, end_time) + elif plugin_name == 'cyberghostvpn_servername': + from vpnservices.cyberghostvpn_servername import CyberghostvpnServername + detector = CyberghostvpnServername(start_time, end_time) + elif plugin_name == 'cyberghostvpn_serverip': + from vpnservices.cyberghostvpn_serverip import CyberghostvpnServerip + detector = CyberghostvpnServerip(start_time, end_time) + elif plugin_name == 'geckovpn_serverip': + from vpnservices.geckovpn_serverip import GeckovpnServerip + detector = GeckovpnServerip(start_time, end_time) else: print('Please input correct plugin name') -- cgit v1.2.3