背景
需求是需要在linux(debian)里运行python程序操作sqlserver读写数据,涉及到数据的读取和写入,这里主要使用pandas进行数据处理。
前提
在运行python程序之前,需要在linux上安装好python和对应的依赖,可以参考以下文章,虽然参考文章都是针对pyodbc连接sqlserver的,但是一开始使用pyodbc的时候,我从数据库读取到的数据一直都是乱码,当我改成pymssql连接的时候就没有乱码了,所以代码中主要是使用pymssql去连接sqlserver。
下面列出一些我参考上面文章用到过的部分命令:
# 解决linux上pip安装pyodbc报错
#src/pyodbc.h:56:10: fatal error: sql.h: No such file or directory
#linux须要安装sqlsrv扩展来连接数据库:
apt-get install unixodbc-dev
apt-get install sqlsrv
# 解决安装tdsodbc报错:debconf: falling back to frontend: Readline:
apt-get -y update && DEBIAN_FRONTEND=noninteractive apt-get -y freetds-dev tdsodbc
#安装pymssql
pip install pymssql
下面列出我安装完依赖服务器上包含的所有包:
# pip list
Package Version
--------------- -------
greenlet 1.0.0
numpy 1.20.2
pandas 1.2.3
pip 21.0.1
pymssql 2.1.5
pyodbc 4.0.30
python-dateutil 2.8.1
pytz 2021.1
scipy 1.6.2
setuptools 54.2.0
six 1.15.0
SQLAlchemy 1.4.3
wheel 0.36.2
实现
编写python文件
# -*-coding:utf-8-*-
import pandas as pd
import numpy as np
import sqlalchemy
import pymssql
from sqlalchemy import create_engine
import json
# 获取数据库连接
engine = create_engine('mssql+pymssql://username:password@192.168.1.1:1433/database?charset=utf8')
# 创建示例数据集
data={"one":np.random.randn(4),"two":np.linspace(1,4,4),"three":['zhangsan','历史',999,0.1]}
df_write=pd.DataFrame(data,index=[1,2,3,4])
# 写入数据库
df_write.to_sql('test', engine, index=False,if_exists='replace',dtype = {
'one': sqlalchemy.types.Float(precision=3, asdecimal=True),
'two': sqlalchemy.types.Float(precision=3, asdecimal=True),
'three': sqlalchemy.types.NVARCHAR(length=255)
})
# 查询语句
sql_query_test = '''select * from test;'''
# 读取到DataFrame
df_test = pd.read_sql_query(sql_query_test, engine)
# 打印读取结果
print(df_test)
运行python文件
python demo.py
查看执行结果
控制台:
数据库: