def get_ports(self, args): ports = [] # Determine list of ports to use if args.port_list is None: if args.low_port is not None or args.high_port is not None: # Try using port low to high port range low = args.low_port high = args.high_port try: if low > 0 and low < high: ports = [i for i in range(low, high + 1)] except: print('Specified port range invalid') else: # Use top 50 ports ports = get_top_50_ports() else: # Use user defined ports all_ports = get_all_ports() user_ports = read_file(args.port_list) # Create dataframe and append description of ports if available descriptions = [None for i in range(len(user_ports))] user_ports = [row['data'] for _, row in user_ports.iterrows()] ports = pd.DataFrame(np.column_stack([user_ports, descriptions]), columns=['port', 'description']) for _, row in ports.iterrows(): new = all_ports.loc[all_ports['port'] == row['port']] if not new.empty: row['description'] = new.iloc[0]['description'] else: row['description'] = 'none' # Return list of ports return ports
def init_kbase(): ''' Read hypernyms from knowledge base ''' names = open_files(KBASE_PATH) kbase = {} for name in names: curr = name.split('/')[-1].split('.')[0] hypernyms = read_file(name, 'txt') kbase[curr] = hypernyms return kbase
def get_fv(cls, dates, labels, news, file_name): path = f'tmp/{file_name}.obj' try: filename = open_file(path) feature_vec = read_file(filename, 'obj') except Exception: type_ = file_name.split('_')[1] feature_vec = FeatureVector(dates, labels, news, type_) save_file(path, feature_vec) return feature_vec
def get_hosts(self, args): hosts = [] # Append all provided hosts to list if args.host is not None: hosts.append(args.host) if args.cidr is not None: nw = ip_network(args.cidr) for ip in nw: hosts.append(str(ip)) if args.ip_list is not None: data = read_file(args.ip_list) for _, row in data.iterrows(): hosts.append(row['data']) # If no hosts are provided, use demo if len(hosts) == 0: hosts.append('scanme.nmap.org') return hosts
from utils.file_io import save_to_file, read_file save_to_file("hi raghu how are u?", "data1.txt") print(read_file("data1.txt"))