首先需要
import pymysql
1.连接数据库
#获取数据库连接
def getDBConn():
conn = pymysql.connect(
host = 'localhost',
port = 3306,
user = 'root',
password = '123',
db = 'insurancedata',
charset = 'utf8'
)
return conn
2.关闭数据库连接
def closeConn(cursor,conn):
if cursor:
cursor.close()
if conn:
conn.close()
3.插入数据
#插入数据
def inset_To_url_table():
try:
conn = getDBConn()
cursor = conn.cursor()
sql = "INSERT INTO url_table(url,url_state)VALUES(%s,%s)"
url = "http://baidu.com/item/123"
status = "0"
params = (url,status)
cursor.execute(sql,params)
conn.commit()
except:
conn.rollback()
print('insert error')
finally:
closeConn(cursor,conn)
4.查询数据
def select_url_table(URL):
try:
conn = getDBConn()
cursor = conn.cursor()
sql = "SELECT url_state from url_table WHERE url = URL"
cursor.execute(sql)
url_state = cursor.fetchall()[0][0]
return url_state
except:
print('select error')
return None
finally:
closeConn(cursor,conn)