def initDeviceListFromDB(self, sqlMana=None): sql = 'select ID from device where roomID = %d' % self.ID res = sqlMana.query(sql) for temp in res: tempDevice = device() tempDevice.readInfoFromDB(temp.get('ID'), sqlMana) self.deviceList.append(tempDevice)
def add2list(pkt): stre = -(256-ord(pkt.notdecoded[-4])) dev_found = False for dev in device_list: if pkt.addr2 == dev.mac_addr: dev.update(stre,get_current_minute(), get_current_time()) dev_found = True break if not dev_found: device_list.append(device(stre, get_current_minute(), get_current_time(), pkt.addr2))
def add2list(pkt): stre = -(256 - ord(pkt.notdecoded[-4])) dev_found = False for dev in device_list: if pkt.addr2 == dev.mac_addr: dev.update(stre, get_current_minute(), get_current_time()) dev_found = True break if not dev_found: device_list.append( device(stre, get_current_minute(), get_current_time(), pkt.addr2))
async def run_client(request, user, password): """ :param request: Equipamento que será realizado a conexão :param user: Login :param password: Senha :return: Se todas as condições forem aceitas, será retornado OK. """ date_time = strftime("%d%m%Y", time.localtime()) if (request[0:3] == 'LDA' or request[0:3] == 'APS'): directory = '/home/***/Backup/BACKUP_'+date_time+'/'+request[0:6]+'_'+ date_time+'/' path = os.path.exists(directory) if request[0:3] == 'APU' or request[0:3] == 'CAB' or request[0:3] == 'RLA': directory = '/home/***/Backup/BACKUP_'+date_time+'/'+request[0:3]+'OLT_'+ date_time+'/' path = os.path.exists(directory) print(request) print (path) if path == True: if request[:11] =='MyHost'or request[:11] == 'MyHost': ##used paramiko connection because my equipment did not accept asyncssh ssh = paramiko.SSHClient() ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) ssh.connect(devices.device(request), port=22, username=user, password=password, look_for_keys='', timeout=200 * 60) connection = ssh.invoke_shell() output='' stdout= connection.recv(64).decode(encoding='utf-8') outFile = open(directory + request + '_' + date_time+'.txt', "w") outFile.write(stdout[:]) commands = devices.commands(request) for p in commands: connection.send(p + '\n') time.sleep(2) if p == 'Show running-config': time.sleep(10) stdout= connection.recv(2500000).decode(encoding='utf-8') output_final= output_final+ stdout outFile = open(directory + request + '_' + date_time+'.txt', "a+") outFile.write(output[:]+ '\n') return 'ok' else: async with asyncssh.connect(devices.device(request), port=22, username=user, password=password, known_hosts=None, client_keys=None) as conn: process = await conn.create_process() ##used to separate citys and equipments if request[0:3] == 'MyHost' or request[0:3] == 'MyHost' or request[0:3] == 'MyHost': output_final= '' output_final= await process.stdout.readuntil(">") ##used because a huawei equipment. outFile = open(directory + request + '_' + date_time+'.txt', "w") commands = devices.commands(request) outFile.write(output+ '\n') for command_send in commands: process.stdin.write(command_send + '\n') await save_output(process, request, command_send) return "ok" else: if request[6:11] == 'MyHost' or request[6:11] == 'MyHost': ##used because a huawei equipment. output_final= await process.stdout.readuntil(">") outFile = open(directory + request + '_' + date_time + '.txt', "w") else: output_final= await process.stdout.readuntil("*") ## used to equipment cisco outFile = open(directory + request + '_' + date_time + '.txt', "w") commands = devices.commands(request) outFile.write(output_final+ '\n') for command_send in commands: print (command_send) if command_send == 'show running-config verbose full': process.stdin.write(command_send + '\n') await save_output(process, request, command_send) else: process.stdin.write(command_send + '\n') await save_output(process, request, command_send) return "ok"
def sys_infinite(out_file, pot, pot_kwargs, dev_kwargs, prog_kwargs, is_plot=True, is_plot_sublat=False, is_spectral=True, e_params=[-0.1, 0.1, 400], **kwargs): if dev_kwargs['is_wrap_finite']: out_file.prnt('\'is_wrap_finite\' cannot be True for an infinite ' + \ 'system. Setting to False\n') dev_kwargs['is_wrap_finite'] = False # Create the dev dev = device(pot=pot, **dev_kwargs) # ------------------------------------------------------------------------ # # Include parameters in the output file for comparison param_dict = {**dev.get_req_params(), **pot.get_req_params()} out_file.prnt_dict(param_dict, is_newline=False) # ------------------------------------------------------------------------ # # Generate leads lead_left = make_lead(dev, 'L', pot=pot, **dev_kwargs) lead_right = make_lead(dev, 'R', pot=pot, **dev_kwargs) if not pot.pot_params['is_const_channel']: smoothing_info = pot.get_pot_smoothing_info(dev.get_xyz(), dev.get_sublat()) for info in smoothing_info: out_file.prnt(info) if is_plot: # Check if we are meant to pass sublat to the plotting function if is_plot_sublat: pot.plot_pot_3D(dev.get_xyz(), dev.get_sublat()) else: pot.plot_pot_3D(dev.get_xyz()) dev.plot_interface(pot.int_loc) dev.plot_energies() param_dict = {**dev.get_req_params(), **pot.get_req_params()} #### LDOS #### energy = 0.0 #plot_LDOS_k_average(lead_left, lead_right, dev, energy) #### SPECTRAL FUNCTION #### if is_spectral: str_ex = None #'_atom_2' start_spectral = time.time() k_num, en_num, spec_data = get_spectral(out_file, lead_left, lead_right, dev, pot_kwargs['gap_val'], **prog_kwargs) out_file.prnt('Time to calculate spectral data : ' + time_elapsed_str(time.time() - start_spectral)) save_spectral(spec_data, dev, pot, k_num, en_num) out_file.prnt('Calculating transmission') en_list = np.linspace(*e_params) k_list = [0] #np.linspace(-np.pi, np.pi, 400) get_transmission(out_file, lead_left, lead_right, dev, pot, en_list, k_list, prog_kwargs)
def initDeviceList(self): # print 'in initDeviceList: ', self.getType() # Livingroom if (self.roomType == 'livingRoom'): livingRoomTV = device(0, 'livingRoomTV', 'TV', 0, 0) self.deviceList.append(livingRoomTV) livingRoomLamp = device(1, 'livingRoomLamp', 'lamp', 0, 0) self.deviceList.append(livingRoomLamp) livingRoomDoor = device(2, 'livingRoomDoor', 'door', 0, 0) self.deviceList.append(livingRoomDoor) livingRoomWindow = device(3, 'livingRoomWindow', 'window', 0, 0) self.deviceList.append(livingRoomWindow) livingRoomComputer = device(4, 'livingRoomComputer', 'computer', 0, 0) self.deviceList.append(livingRoomComputer) livingRoomCharger = device(5, 'livingRoomCharger', 'charger', 0, 0) self.deviceList.append(livingRoomCharger) livingRoomAirCondition = device(6, 'livingRoomAirCondition', 'airCondition', 0, 25) self.deviceList.append(livingRoomAirCondition) livingRoomAudioSystem = device(7, 'livingRoomAudioSystem', 'other', 0, 0) self.deviceList.append(livingRoomAudioSystem) # Master Bedroom if (self.roomType == 'masterBedroom'): masterBedRoomTV = device(100, 'masterBedRoomTV', 'TV', 0, 0) self.deviceList.append(masterBedRoomTV) masterBedRoomCeilingLamp = device(101, 'masterBedRoomCeilingLamp', 'lamp', 0, 0) self.deviceList.append(masterBedRoomCeilingLamp) masterBedRoomBedLamp = device(102, 'masterBedRoomBedLamp', 'lamp', 0, 0) self.deviceList.append(masterBedRoomBedLamp) masterBedRoomDoor = device(103, 'masterBedRoomDoor', 'door', 0, 0) self.deviceList.append(masterBedRoomDoor) masterBedRoomWindow = device(104, 'masterBedRoomWindow', 'window', 0, 0) self.deviceList.append(masterBedRoomWindow) masterBedRoomAirCondition = device(105, 'masterBedRoomAirCondition', 'airCondition', 0, 25) self.deviceList.append(masterBedRoomAirCondition) masterBedRoomComputer = device(106, 'masterBedRoomComputer', 'computer', 0, 0) self.deviceList.append(masterBedRoomComputer) masterBedRoomCharger = device(107, 'masterBedRoomCharger', 'charger', 0, 0) self.deviceList.append(masterBedRoomCharger) # Second Bedroom if (self.roomType == 'secondBedroom'): secondBedRoomTV = device(200, 'secondBedRoomTV', 'TV', 0, 0) self.deviceList.append(secondBedRoomTV) secondBedRoomCeilingLamp = device(201, 'secondBedRoomCeilingLamp', 'lamp', 0, 0) self.deviceList.append(secondBedRoomCeilingLamp) secondBedRoomBedLamp = device(202, 'secondBedRoomBedLamp', 'lamp', 0, 0) self.deviceList.append(secondBedRoomBedLamp) secondBedRoomDoor = device(203, 'secondBedRoomDoor', 'door', 0, 0) self.deviceList.append(secondBedRoomDoor) secondBedRoomWindow = device(204, 'secondBedRoomWindow', 'window', 0, 0) self.deviceList.append(secondBedRoomWindow) secondBedRoomAirCondition = device(205, 'secondBedRoomAirCondition', 'airCondition', 0, 25) self.deviceList.append(secondBedRoomAirCondition) secondBedRoomComputer = device(206, 'secondBedRoomComputer', 'computer', 0, 0) self.deviceList.append(secondBedRoomComputer) secondBedRoomCharger = device(207, 'secondBedRoomCharger', 'charger', 0, 0) self.deviceList.append(secondBedRoomCharger) # Third Bedroom if (self.roomType == 'thirdBedroom'): thirdBedRoomTV = device(300, 'thirdBedRoomTV', 'TV', 0, 0) self.deviceList.append(thirdBedRoomTV) thirdBedRoomCeilingLamp = device(301, 'thirdBedRoomCeilingLamp', 'lamp', 0, 0) self.deviceList.append(thirdBedRoomCeilingLamp) thirdBedRoomBedLamp = device(302, 'thirdBedRoomBedLamp', 'lamp', 0, 0) self.deviceList.append(thirdBedRoomBedLamp) thirdBedRoomDoor = device(303, 'thirdBedRoomDoor', 'door', 0, 0) self.deviceList.append(thirdBedRoomDoor) thirdBedRoomWindow = device(304, 'thirdBedRoomWindow', 'window', 0, 0) self.deviceList.append(thirdBedRoomWindow) thirdBedRoomAirCondition = device(305, 'thirdBedRoomAirCondition', 'airCondition', 0, 25) self.deviceList.append(thirdBedRoomAirCondition) thirdBedRoomComputer = device(306, 'thirdBedRoomComputer', 'computer', 0, 0) self.deviceList.append(thirdBedRoomComputer) thirdBedRoomCharger = device(307, 'thirdBedRoomCharger', 'charger', 0, 0) self.deviceList.append(thirdBedRoomCharger) # Bathroom if (self.roomType == 'bathroom'): bathroomLamp = device(400, 'bathroomLamp', 'lamp', 0, 0) self.deviceList.append(bathroomLamp) bathroomWaterHeater = device(401, 'bathroomWaterHeater', 'heater', 0, 0) self.deviceList.append(bathroomWaterHeater) bathroomDoor = device(402, 'bathroomDoor', 'door', 0, 0) self.deviceList.append(bathroomDoor) bathroomWindow = device(403, 'bathroomWindow', 'window', 0, 0) self.deviceList.append(bathroomWindow) bathroomCharger = device(404, 'bathroomCharger', 'charger', 0, 0) self.deviceList.append(bathroomCharger) bathroomFan = device(405, 'bathroomFan', 'other', 0, 0) self.deviceList.append(bathroomFan) # Kitchen if (self.roomType == 'kitchen'): kitchenLamp = device(500, 'kitchenLamp', 'lamp', 0, 0) self.deviceList.append(kitchenLamp) kitchenDoor = device(501, 'kitchenDoor', 'door', 0, 0) self.deviceList.append(kitchenDoor) kitchenWindow = device(502, 'kitchenWindow', 'window', 0, 0) self.deviceList.append(kitchenWindow) kitchenVentilator = device(503, 'kitchenVentilator', 'other', 0, 0) self.deviceList.append(kitchenVentilator) # Dining Room if (self.roomType == 'diningRoom'): diningRoomLamp = device(600, 'diningRoomLamp', 'lamp', 0, 0) self.deviceList.append(diningRoomLamp) diningRoomDoor = device(601, 'diningRoomDoor', 'door', 0, 0) self.deviceList.append(diningRoomDoor) diningRoomFan = device(602, 'diningRoomFan', 'other', 0, 0) self.deviceList.append(diningRoomFan) diningRoomWindow = device(603, 'diningRoomWindow', 'window', 0, 0) self.deviceList.append(diningRoomWindow) diningRoomAirCondition = device(604, 'diningRoomAirCondition', 'airCondition', 0, 0) self.deviceList.append(diningRoomAirCondition) # Study Room if (self.roomType == 'studyRoom'): studyRoomCeilingLamp = device(700, 'studyRoomCeilingLamp', 'lamp', 0, 0) self.deviceList.append(studyRoomCeilingLamp) studyRoomDeskLamp = device(701, 'studyRoomDeskLamp', 'lamp', 0, 0) self.deviceList.append(studyRoomDeskLamp) studyRoomAirCondition = device(702, 'studyRoomAirCondition', 'airCondition', 0, 0) self.deviceList.append(studyRoomAirCondition) studyRoomDoor = device(703, 'studyRoomDoor', 'door', 0, 0) self.deviceList.append(studyRoomDoor) studyRoomWindow = device(704, 'studyRoomWindow', 'window', 0, 0) self.deviceList.append(studyRoomWindow) studyRoomComputer = device(705, 'studyRoomComputer', 'computer', 0, 0) self.deviceList.append(studyRoomComputer) studyRoomCharger = device(706, 'studyRoomCharger', 'charger', 0, 0) self.deviceList.append(studyRoomCharger) studyRoomAudioSystem = device(707, 'studyRoomAudioSystem', 'other', 0, 0) self.deviceList.append(studyRoomAudioSystem) self.setDevicesPosRandom()
async def run_client(request, user, password): date_time = strftime("%d%m%Y", time.localtime()) if (request[0:3] == 'LDA' or request[0:3] == 'APS'): dir = '/home/data/Backup/BACKUP_' + date_time + '/' + request[ 0:6] + '_' + date_time + '/' path = os.path.exists(dir) if request[0:3] == 'APU' or request[0:3] == 'CAB' or request[0:3] == 'RLA': dir = '/home/data/Backup/BACKUP_' + date_time + '/' + request[ 0:3] + 'OLT_' + date_time + '/' path = os.path.exists(dir) if path == True: if request[:11] == 'APSDTCRTD01' or request[:11] == 'LDADTCCMT06': ssh = paramiko.SSHClient() ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) ssh.connect(devices.device(request), port=22, username=user, password=password, look_for_keys='', timeout=200 * 60) connection = ssh.invoke_shell() output = '' stdout = connection.recv(64).decode(encoding='utf-8') outFile = open(dir + request + '_' + date_time + '.txt', "w") outFile.write(stdout[:]) comandos = devices.comandos(request) for p in comandos: connection.send(p + '\n') time.sleep(2) if p == 'Show running-config': time.sleep(10) stdout = connection.recv(2500000).decode(encoding='utf-8') output = output + stdout print(output[:]) outFile = open(dir + request + '_' + date_time + '.txt', "a+") outFile.write(output[:] + '\n') return 'ok' else: async with asyncssh.connect(devices.device(request), port=22, username=user, password=password, known_hosts=None, client_keys=None) as conn: process = await conn.create_process() print("passei") if request[0:3] == 'APU' or request[0:3] == 'CAB' or request[ 0:3] == 'RLA': output = '' output = await process.stdout.readuntil(">") outFile = open(dir + request + '_' + date_time + '.txt', "w") comandos = devices.comandos(request) outFile.write(output + '\n') for cmd in comandos: process.stdin.write(cmd + '\n') await save_output(process, request, cmd) return "ok" else: if request[6:11] == 'RTD03' or request[6:11] == 'RTD04': output = await process.stdout.readuntil(">") outFile = open( dir + request + '_' + date_time + '.txt', "w") elif request[6:11] == 'CMT06': output = await process.stdout.readuntil(request) outFile = open( dir + request + '_' + date_time + '.txt', "w") else: output = await process.stdout.readuntil("*") outFile = open( dir + request + '_' + date_time + '.txt', "w") comandos = devices.comandos(request) outFile.write(output + '\n') for cmd in comandos: print(cmd) if cmd == 'show running-config verbose full': process.stdin.write(cmd + '\n') await save_output(process, request, cmd) else: process.stdin.write(cmd + '\n') await save_output(process, request, cmd) return "ok" else: return "Pasta não está criada!."