def __init__(self): self.ipList = [] query=("SELECT " "asset.id," "asset.name," "node_ip_mapping.ip " "FROM " "node_ip_mapping " "left join asset on asset.id = node_ip_mapping.asset_id " "WHERE node_ip_mapping.status = 'ACTIVE' " "AND asset.status = 'ACTIVE' " "AND asset.id = %s") queryArg = ('abc',) logger.info('Fetch fix ip') database = fia() result = database.lookup(query,queryArg) for (ip) in result: self.ipList.append(ip)
def isEmergency(self,name): magic_number = 4953 query=("SELECT DISTINCT " "booking.id from booking " "left join booking_stakeholder on booking_stakeholder.booking_id =booking.id " "left join fiauser on fiauser.id = booking_stakeholder.user_id " "WHERE booking.status = 'ACTIVE' " " AND fiauser.signum = %s ") queryArg = (name,) database = fia() result = database.lookup(query,queryArg) bookingList = [] for (bookingId,) in result: bookingList.append(bookingId) try: bookingList.index(magic_number) return True except: return False
def __init__(self,assetid,nodename): self.ipList = [] query=("SELECT " "asset.id," "asset.name," "node_ip_mapping.ip " "FROM " "node_ip_mapping " "left join asset on asset.id = node_ip_mapping.asset_id " "WHERE node_ip_mapping.status = 'ACTIVE' " "AND asset.status = 'ACTIVE' " "AND asset.id = %s") queryArg = (assetid,) logger.info('Lookup ip for ' + nodename) database = fia() result = database.lookup(query,queryArg) for (asset_id, asset_name, node_ip) in result: self.ipList.append(node_ip)
def __init__(self,name): ipListAll = [] query=("SELECT DISTINCT " "fiauser.signum," "asset.id," "asset.name " "FROM " "booking_stakeholder " "left join fiauser on fiauser.id = booking_stakeholder.user_id " "left join assignment on assignment.booking_id = booking_stakeholder.booking_id " "left join asset on asset.id = assignment.asset_id " "left join booking on booking.id = assignment.booking_id " "WHERE 1=1 " "AND assignment.status = 'ACTIVE' " "AND booking.status = 'ACTIVE' " "AND fiauser.signum = %s ") queryArg = (name,) logger.info('Lookup booking information for ' + name) database = fia() result = database.lookup(query,queryArg) nodeList = [] for (username, assetid, assetname) in result: item = node(assetid,assetname) nodeList.append(item) for item in nodeList: ipListAll += item.ipList #Dirty code to add immune IP ipListAll.append('127.0.0.1') #Fetch fixip query=("SELECT * FROM fia.fixed_ip ") queryArg = () logger.info('Lookup fix IP for ' + name) result = database.lookup(query,queryArg) fixIPList = [] for (id, ip, comment) in result: fixIPList.append(ip) ipListAll += fixIPList self.ipList = list(set(ipListAll))
def test(): database = fia() query=("SELECT DISTINCT " "fiauser.signum," "asset.id," "asset.name " "FROM booking_stakeholder " "left join fiauser on fiauser.id = booking_stakeholder.user_id " "left join assignment on assignment.booking_id = booking_stakeholder.booking_id " "left join asset on asset.id = assignment.asset_id " "left join booking on booking.id = assignment.booking_id " "WHERE 1=1 " "AND assignment.status = 'ACTIVE' " "AND booking.status = 'ACTIVE' " "AND fiauser.signum = 'enemali' ") queryArg=() lines = database.lookup(query, queryArg) try: assert str(lines[0]) == "('enemali', 1594, 'RBS72-6601-W')" print('database.fia.lookup ...... pass') except: print('database.fia.lookup ...... failed') print('Debug info:') print(lines[0])