cross_doctor/app/tmnet_db_func.py

from app.db_func_base import TableDbHelperBase


class TmnetDbHelper(TableDbHelperBase):

    def __init__(self, pool):
        self.db_pool = pool
        self.DB_Name = 'tmnet'

    def reload_cross_info(self, nodeid, last_reload_time):
        sql1 = "select * from cross_ledger_update_info where nodeid = %s and crossid is not null" % nodeid
        sql2 = "select * from road_ledger_update_info where nodeid = %s" % nodeid
        crosses_list, cross_res_list = self.do_select(sql1), []
        for cross_row in crosses_list:
            if cross_row['update_time'] > last_reload_time:
                cross_res_list.append(cross_row)
        roads_list, road_res_list = self.do_select(sql2), []
        for road_row in roads_list:
            if road_row['update_time'] > last_reload_time:
                road_res_list.append(road_row)
        return cross_res_list, road_res_list
    
    def query_phase_by_cross_list(self, citycode, cross_list):
        conn, cursor = self.connect()
        try:
            sql = f'''select
                        cm.xl_crossid
                    from phasetable.cross_mapping as cm
                    join phasetable.day_schedule as ds on cm.jj_crossid = ds.crossid and cm.nodeid = cm.nodeid
                    where
                        cm.nodeid = %s and cm.xl_crossid in %s
                    group by cm.xl_crossid'''
            print(cursor.mogrify(sql,(citycode,cross_list)))
            cursor.execute(sql,(citycode,cross_list))
            result = cursor.fetchall()
            self.close(conn, cursor)
            return result, None
        except Exception as error:
            self.close(conn, cursor)
            return None, error

    def query_cross_info(self, nodeid):
        sql = "select * from `cross` where nodeid = %s" % nodeid
        return self.do_select(sql)

    def query_cross_info_new(self, citycode):
        conn, cursor = self.connect()
        try:
            sql = f'''select * from `cross` where nodeid = %s'''
            print(cursor.mogrify(sql, (citycode)))
            cursor.execute(sql, (citycode))
            result = cursor.fetchall()
            self.close(conn, cursor)
            return result, None
        except Exception as error:
            self.close(conn, cursor)
            return None, error

    def query_cross_list_sql(self, nodeid, area_id):
        sql = """
            select 
                   if(t2.name is not null, t2.name, t1.name) as name,
                   t1.crossid,
                   if (t2.location is not null, t2.location, t1.location) as location,
                   t1.nodeid,
                   t1.area_id
                from `cross` as t1
                left join cross_ledger_update_info as t2 on t1.crossid=t2.crossid
                where t1.at_edge=0 and t1.nodeid = %s and t1.area_id = %s
        """ % (nodeid, area_id)
        return self.do_select(sql)

    def query_cross_inroads(self, crossid, nodeid):
        sql = """
            select
                t1.roadid,                     
                if(t2.from_crossid is not null, t2.from_crossid, t1.from_crossid) as from_crossid,
                if(t2.to_crossid is not null, t2.to_crossid, t1.to_crossid) as to_crossid,
                if(t2.name is not null, t2.name, t1.name) as name,
                if(t2.src_direct is not null, t2.src_direct, t1.src_direct) as src_direct
            from
            (select * from road where nodeid = '%s' and recordstate=0 and to_crossid='%s' and (is_sup_road is null or is_sup_road<>1)) t1
            left join
            (select * from road_ledger_update_info where nodeid = '%s' and recordstate=0 and to_crossid='%s' and (is_sup_road is null or is_sup_road<>1)) t2
            on t1.roadid = t2.roadid;
        """ % (nodeid, crossid, nodeid, crossid)
        return self.do_select(sql)

    def query_base_info(self, nodeid, area_id):
        sql = "select * from ledger.leger_base_info where nodeid= %d and area_id = %s" % (int(nodeid), int(area_id))
        return self.do_select(sql)