def getJobCount(self): try: super(JobDAO, self).getConnection() sql_c = "select count from t_c_job" result_c = super(JobDAO, self).fetchall(sql_c) sql_java = "select count from t_java_job" result_java = super(JobDAO, self).fetchall(sql_java) sql_py = "select count from t_py_job" result_py = super(JobDAO, self).fetchall(sql_py) sql_ai = "select count from t_ai_job" result_ai = super(JobDAO, self).fetchall(sql_ai) sql_web = "select count from t_qd_job" result_web = super(JobDAO, self).fetchall(sql_web) super(JobDAO, self).commit() return result_c, result_java, result_py, result_ai, result_web pass except Exception as e: logger.error("执行SQL:" + sql_c + "出现异常:" + str(e)) pass finally: pass
def executemany(self, sql, param): try: if self.__cursor and self.__conn: self.__cursor.executemany(sql, param) except: logger.error("执行语句:" + sql + str(param)) self.__cursor.close() self.__conn.close()
def getConnection(self): try: self.__conn = pymysql.connect(self.__host, self.__name, self.__pwd,self.__dbname,charset=self.__charset) except (pymysql.MySQLError, pymysql.DatabaseError, Exception): logger.error("数据库连接异常:" + self.__host) pass self.__cursor = self.__conn.cursor() pass
def execute(self, sql, param=None): try: if self.__conn and self.__cursor: if param: self.__cursor.execute(sql, param) else: self.__cursor.execute(sql) except: logger.error("执行语句:" + sql, str(param)) self.__cursor.close() self.__conn.close()
def getAllUsers(self): try: super().getConnection() sqlSelect = "select * from t_user " result = super().fetchall(sqlSelect) super().commit() return result except Exception as e: logger.error("执行SQL:" + sqlSelect + " 出现异常,params:" + str(e) ) finally: super().close()
def getmysqlport(self): try: super().getConnection() sqlSelect = "select * from t_port where port = '3306'" result = super().fetchall(sqlSelect) super().commit() return result except Exception as e: logger.error("执行SQL:" + sqlSelect + " 出现异常,params:" + str(e) ) finally: super().close()
def insertvultotal(self, ip, danger): try: super().getConnection() sqlinsersqlvul = "update t_total set %s=%s + 1 where ip=%s"%(danger,danger,ip) result = super().execute(sqlinsersqlvul) super().commit() return result except Exception as e: logger.error("执行SQL:" + sqlSelect + " 出现异常,params:" + str(e)) finally: super().close()
def getConnection(self): try: self.__conn = pymysql.connect(self.__host, self.__username, self.__password, self.__schema, charset=self.__charset) except (pymysql.DatabaseError, pymysql.MySQLError, Exception): logger.error("数据库连接失败。。。 " + self.__host) self.__cursor = self.__conn.cursor()
def getpriescvul(self, prilist): try: super().getConnection() sqlSelect = "select * from t_vul where ip = %s and typeid = '3'" params = (prilist) result = super().fetchall(sqlSelect, params) super().commit() return result except Exception as e: logger.error("执行SQL:" + sqlSelect + " 出现异常,params:" + str(e)) finally: super().close()
def getsshpass(self,idpass): try: super().getConnection() sqlSelect = "select * from t_weakpass where typeid = %s" params = (idpass) result = super().fetchall(sqlSelect,params) super().commit() return result except Exception as e: logger.error("执行SQL:" + sqlSelect + " 出现异常,params:" + str(e) ) finally: super().close()
def getport(self,portlist): try: super().getConnection() sqlSelect = "select * from t_port where ip = %s" params = (portlist) result = super().fetchall(sqlSelect,params) super().commit() return result except Exception as e: logger.error("执行SQL:" + sqlSelect + " 出现异常,params:" + str(e) ) finally: super().close()
def getJobYw(self): try: super().getConnection() sql = "select maxsalary,minsalary,avgsalary,jobcount from t_yw" result = super().fetchall(sql) super().commit() return result pass except Exception as e: logger.error() finally: super().close()
def getUserinfo(self,name): try: super().getConnection() sqlSelect = "select userpwd from t_user where username = %s" params = (name) result = super().fetchall(sqlSelect,params) super().commit() return result except Exception as e: logger.error("执行SQL:" + sqlSelect + " 出现异常,params:" + str(e)) finally: super().close()
def getJobWeb(self): try: super(JobDAO, self).getConnection() sql = "select max,min,avg from t_qd_job" result = super(JobDAO, self).fetchall(sql) super(JobDAO, self).commit() return result pass except Exception as e: logger.error("执行SQL:" + sql + " 出现异常:" + str(e)) finally: super(JobDAO, self).close()
def getJobCounts(self): try: super().getConnection() sql = "select count(*) from t_job_info" result = super().fetchall(sql) super().commit() # print(result) return result except Exception as e: logger.error("执行SQL:" + sql + " 出现异常,params:" + " " + str(e)) finally: super().close()
def executeQuery(self, sql, param, row='one'): try: if self.__cursor and self.__conn: self.__cursor.execute(sql, param) if row == 'one': return self.__cursor.fetchone() else: return self.__cursor.fetchall() except: logger.error("执行语句:" + sql) self.__cursor.close() self.__conn.close()
def insertports(self,ip,port,protocol,status,service): # ip, port, protocol, status, service try: super().getConnection() sqlinsertport = "insert into t_port (ip,port,protocol,status,service) values (%s,%s,%s,%s,%s)" params = (ip,port,protocol,status,service) result = super().execute(sqlinsertport, params) super().commit() return result except Exception as e: logger.error("执行SQL:" + sqlSelect + " 出现异常,params:" + str(e) ) finally: super().close()
def deleteUserById(self, userId): try: super().getConnection() sqlDelete= "delete from t_user where userId=%s" params = (userId,) result = super().execute(sqlDelete, params) super().commit() return result except Exception as e: super().rollback() logger.error("执行SQL:" + sqlDelete + " 出现异常,params:" + params + str(e)) finally: super().close()
def getUserByUserNameAndPwd(self, user): try: super().getConnection() sqlSelect = "select roleid,username, userpwd from t_user where username=%s and userpwd=%s" params = (user.userName, user.userPwd) result = super().fetchone(sqlSelect, params) print(result) super().commit() return result except Exception as e: logger.error("执行SQL:" + sqlSelect + " 出现异常,params:" + params + " " + str(e) ) finally: super().close()
def getLimitJobCitys(self, city, currentPage, pageNO): try: super().getConnection() sql = "select * from t_job_info where jobCity = %s limit %s,%s" params = (city, (currentPage - 1) * pageNO, pageNO) result = super().fetchall(sql, params) super().commit() # print(result) return result except Exception as e: logger.error("执行SQL:" + sql + " 出现异常,params:" + " " + str(e)) finally: super().close()
def inserthosts(self,system,accuracy,Host): # system,accuracy try: super().getConnection() sqlinserthost = "insert into t_host (system,accuracy,ip) values (%s,%s,%s)" params = (system,accuracy,Host) result = super().execute(sqlinserthost, params) super().commit() return result except Exception as e: logger.error("执行SQL:" + sqlSelect + " 出现异常,params:" + str(e) ) finally: super().close()
def getLimitJobPosition(self, position, currentPage, pageNO): try: super().getConnection() sql = "select * from t_job_info where jobPosition like '%{0}%' limit {1},{2}".format( position, (currentPage - 1) * pageNO, pageNO) result = super().fetchall(sql) super().commit() # print(result) return result except Exception as e: logger.error("执行SQL:" + sql + " 出现异常,params:" + " " + str(e)) finally: super().close()
def getJobNumCity(self, city): try: super().getConnection() sql = "select count(*) from t_job_info where jobCity = %s" params = (city, ) result = super().fetchall(sql, params) super().commit() # print(result) return result except Exception as e: logger.error("执行SQL:" + sql + " 出现异常,params:" + " " + str(e)) finally: super().close()
def insertsshweak(self,idpass,host,weaktype,weakuser,weakpwd): # idpass,weaktype,weakuser,weakpwd try: super().getConnection() sqlinsertip = "insert into t_weakpass (idpass,ip,weaktype,weakuser,weakpass) values (%s,%s,%s,%s,%s)" params = (idpass,host,weaktype,weakuser,weakpwd) result = super().execute(sqlinsertip, params) super().commit() return result except Exception as e: logger.error("执行SQL:" + sqlSelect + " 出现异常,params:" + str(e) ) finally: super().close()
def getJobNumPosition(self, position): try: super().getConnection() sql = "select count(*) from t_job_info where jobPosition like '%{0}%'".format( position) result = super().fetchall(sql) super().commit() # print(result) return result except Exception as e: logger.error("执行SQL:" + sql + " 出现异常,params:" + " " + str(e)) finally: super().close()
def setUserTelByName(self, user, telvalue): try: super().getConnection() sql = "update t_user set usertel = '{0}' where username = '******'".format( telvalue, user.username) result = super().execute(sql) super().commit() # print(result) return result except Exception as e: logger.error("执行SQL:" + sql + " 出现异常,params:" + " " + str(e)) finally: super().close()
def getLimitJobByMaxSalary(self, currentPage, pageNO): try: super().getConnection() sql = "select * from t_job_info order by jobMaxsalary desc limit %s,%s" params = (currentPage, pageNO) result = super().fetchall(sql, params) super().commit() # print(result) return result except Exception as e: logger.error("执行SQL:" + sql + " 出现异常,params:" + " " + str(e)) finally: super().close()
def getnmapinfo(self, vullist): try: super().getConnection() vulSelect = "select domain_name from t_ip where ip = %s" params = (vullist) result = super().fetchall(vulSelect, params) print(result) super().commit() return result except Exception as e: logger.error("执行SQL:" + sqlSelect + " 出现异常,params:" + str(e)) finally: super().close()
def insertpaths(self,ip,domainname,path,state): # system,accuracy try: super().getConnection() sqlinsertpath = "insert into t_path (ip,domainname,path,state) values (%s,%s,%s,%s)" params = (ip,domainname,path,state) result = super().execute(sqlinsertpath, params) super().commit() return result except Exception as e: logger.error("执行SQL:" + sqlSelect + " 出现异常,params:" + str(e) ) finally: super().close()
def insertips(self,ip,domianname): # ip,domianname try: super().getConnection() sqlinsertip = "insert into t_ip (ip,domain_name) values (%s,%s)" params = (ip,domianname) result = super().execute(sqlinsertip, params) super().commit() return result except Exception as e: logger.error("执行SQL:" + sqlSelect + " 出现异常,params:" + str(e) ) finally: super().close()