python脚本获取数据库数据并输出为python对象

假设应用:某接口需要从数据库获取数据,每条数据都是一个python对象,以数组形式输出,即: [{}, {}, {}…],以python2.7和sqlserver为例

测试数据:
在这里插入图片描述
python脚本:

# coding=utf-8
import pyodbc

#数据库连接
conn = pyodbc.connect('DRIVER={SQL Server};SERVER=20.209.182.190;DATABASE=testdb;UID=test;PWD=test@123')
cur = conn.cursor()

#各字段空数组
ipList = []  #ip数组
cpuList = [] #cpu数组 
diskList = [] #硬盘数组
ramList = [] #内存数组
objList = [] #对象数组

sql = "select * from A_TEST" # 查询所有数据
cur.execute(sql)
for row in cur.fetchall():
    #将各个字段数据放入相应的数组
    ipList.append(row[0])
    cpuList.append(row[1])
    diskList.append(row[2])
    ramList.append(row[3])

sql_count = "select count(1) from A_TEST"  #查询有多少条数据
cur.execute(sql_count)
for count in cur.fetchall():
    num = int(count[0]) #获取数据条数并转为int类型
  
for i in range(0,num): #循环每条数据
    obj = {"ip":ipList[i], "cpu_use":cpuList[i], "disk_use":diskList[i],"ram_use":ramList[i]} #取每个字段数据组成对象
    objList.append(obj) #将对象放入对象数组

print objList #打印对象数组

print (u"第二条数据ip信息: " + objList[1]['ip']) #打印第二条数据的ip信息
print (u"第三条数据cpu使用率信息: " + objList[2]['cpu_use']) #打印第三条数据的cpu使用率信息   

conn.close() #关闭数据库连接


运行结果:
在这里插入图片描述

  • 0
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
在NiFi中使用ExecuteScript执行Python脚本获取数据库连接,执行SQL,可以按照以下步骤进行操作: 1. 在NiFi中创建一个ExecuteScript处理器,并选择要执行的脚本语言为Python。 2. 编写Python脚本获取数据库连接,执行SQL并返回结果。以下是一个简单的示例: ```python from org.apache.nifi.processor.io import StreamCallback from org.apache.nifi.processor.util import StandardCharsets from org.apache.nifi.processor import * from java.io import BufferedReader, InputStreamReader import java.sql class PyStreamCallback(StreamCallback): def __init__(self): pass def process(self, inputStream, outputStream): connection = java.sql.DriverManager.getConnection('jdbc:mysql://localhost:3306/test', 'username', 'password') statement = connection.createStatement() resultSet = statement.executeQuery('SELECT * FROM table') while resultSet.next(): value = resultSet.getString('column_name') outputStream.write(bytearray(value, StandardCharsets.UTF_8)) flowFile = session.get() if flowFile != None: flowFile = session.write(flowFile, PyStreamCallback()) session.transfer(flowFile, REL_SUCCESS) ``` 在这个脚本中,我们首先导入了需要使用的Java类,包括StreamCallback、StandardCharsets、Processor等。然后定义了一个名为PyStreamCallback的类,继承自StreamCallback,用于处理输入流和输出流。 在process方法中,我们使用Java的JDBC API获取数据库连接,执行SQL语句,并将结果输出输出流中。最后,我们使用NiFi的API将处理后的流传输到下一个处理器。 3. 配置ExecuteScript处理器的属性,包括Python解释器路径、脚本文件路径等。 4. 运行NiFi流程,执行Python脚本获取数据库连接,执行SQL并返回结果。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值