Exemplo n.º 1
0
    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)
Exemplo n.º 2
0
 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
Exemplo n.º 3
0
    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)
Exemplo n.º 4
0
    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))
Exemplo n.º 5
0
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])