python使用Pymysql连接数据库并写入数据

本文主要讲述python使用Pymysql连接数据库并写入数据

1.首先讲连接数据库

#!/usr/bin/python
# -*- coding: UTF-8 -*-

#首先导入模块pymysql,若出现报错没有此模块，就pip install pymysql
import os,sys,pymysql

#使用cursor()方法创建游标对象cursor
cursor = db.cursor()

#使用execute()方法执行sql查询当前版本
cursor.execute('select version()')

#使用fetchone()方法获取单条数据
data = cursor.fetchone()
print ("Database version : %s " % data)



# 关闭游标
cursor.close()
# 关闭数据库连接
db.close()

2.用python执行shell命令，调用os模块

def get_ip():
     #使用os.popen模块执行shell命令查看本机ip,注意外围使用双引号而非单引号,并且假设默认是第一个网卡,特殊环境请适当修改代码  
     #out = os.popen("ifconfig enp0s3 | grep 'inet '|awk '{print $2}'").read()
     out = os.popen("ifconfig|grep -oP '(?<=inet addr:)(?=(?!127\.0\.0\.1))\d+(\.\d+){3}'").read()
     out = out.strip() #此步骤是因为去头尾多余的空格，以免插入数据库出现格式问题；
     print (out) 

 

3.用python执行shell脚本，调用os.system

# 使用 os.system执行shell脚本

p = os.system("../shell/rtc-server-monitor-log.sh")

4.完整代码

#!/usr/bin/python
# -*- coding: UTF-8 -*-

#首先导入模块pymysql,若出现报错没有此模块，就pip install pymysql
import os,sys,pymysql
def get_ip():
     #使用os.popen模块执行shell命令查看本机ip,注意外围使用双引号而非单引号,并且假设默认是第一个网卡,特殊环境请适当修改代码  
     #out = os.popen("ifconfig enp0s3 | grep 'inet '|awk '{print $2}'").read()
     out = os.popen("ifconfig|grep -oP '(?<=inet addr:)(?=(?!127\.0\.0\.1))\d+(\.\d+){3}'").read()
     out = out.strip() #此步骤是因为去头尾多余的空格，以免插入数据库出现格式问题；
     # print (out) 
     res = out
     return res
# 使用 os.system执行shell脚本

p = os.system("../shell/rtc-server-monitor-log.sh")

#print(p")


#使用cursor()方法创建游标对象cursor
cursor = db.cursor()

#使用execute()方法执行sql查询当前版本
cursor.execute('select version()')

#使用fetchone()方法获取单条数据
data = cursor.fetchone()
print ("Database version : %s " % data)


#T = []
#for num  in range(p)
# T.append((num))  # 注意要用两个括号括起来
#sql插入语句：
#sql = """INSERT INTO rtc_excep_task(ip,任务id,告警类型)
#         VALUES ('10.114.123.45','31184','1')"""
#open_file = open('/tmp/rtc-server-log-fail')
#ith open("/tmp/rtc-server-log-fail") as f

#运行前清空数据表；
cursor.execute('delete from rtc_excep_task')
print('清空表rtc_excep_task完成！')
#数据库插入语句
sql = """INSERT INTO rtc_excep_task(IP,TASKID,DATE,ERRORSYSTEM)
         VALUES (%s,%s,%s,%s)"""
 

try:
#读取linux文件并且取字段值
  with open("/tmp/rtc-server-log-fail") as f:
       
        for line in f.readlines():
           line = line.strip() #去掉每行头尾空白
           _nodes = line.split(' ')#以空格为分隔符提取字段
           id = _nodes[1]
           DATE = _nodes[0]
           #print(id)
           #print(DATE)
           system = "1"
           #out = get_ip()
           ipl = []
           #ip = out
           ip = get_ip()
           ipl.append(ip)
           print(ipl)
           #print(type(ipl))
        # 执行sql语句
           cursor.execute(sql,(ip,id,DATE,system))
         # 提交到数据库执行
           db.commit()
           print ("数据库插入成功！")
         # count += 1
except Exception as e :
        print(e)
        # 如果发生错误则回滚
        db.rollback()
        print("数据库插入不成功，已回滚！")
# 关闭游标
cursor.close()
# 关闭数据库连接
db.close()

运行结果：

[appdeploy@CNSZ17VLK0922:/app/anaconda2/bin]$/app/anaconda2/bin/python /app/python/rtc-monitor-server.py 
清空表rtc_excep_task完成！
['192.168.56.33']
数据库插入成功！