#!/usr/bin/env python
# -*- coding: utf-8 -*-
__author__ = 'yangshijin'
# coding=utf-8
""" 测试代码(mysql存储文件) """
import pymysql
# TODO 需要z自行替换的字段
DBName = 'model_save' # 数据库
tabel_name = 'models' # 数据库_表格
columnName = 'xxxx' # 存储文件字段名称,类型:longblob
inDoc = './py4.jpg' # 需要存入数据库的文件路径
#inDoc = './沧海一声笑.mp3' # 需要存入数据库的文件路径
outDoc = './py4_dump.jpg' # 从数据库读取后恢复的文件路径
#outDoc = './沧海一声笑_dump_1.mp3' # 从数据库读取后恢复的文件路径
info = 'py_img4'
#info = '沧海一声笑'
def connect():
""" 连接数据库 """
conn = pymysql.connect('localhost',
'root',
'eve136712',
db=DBName,
charset='utf8',
use_unicode=True)
return conn
def openFile(fileName):
""" 打开文件,读取二进制格式,返回处理过的字符串 """
with open(fileName, 'rb') as f:
#print(f.read())
content = 's'.join([str(ele) for ele in f.read()])
f.close()
return content or ''
def dealDBFileData(data):
""" 对从数据库fetchall的二进制进行处理 """
temp = data[0][1].decode('ascii').split('s')
return bytes([int(ele) for ele in temp])
if __name__ == '__main__':
conn = connect()
cursor = conn.cursor()
# 打开文件,读取二进制
text = openFile(inDoc)
#===========将数据存储到数据库===========
insert_sql="INSERT INTO models (model, info) VALUES (%s, %s)"
if cursor.execute(insert_sql , (text, info)):
conn.commit()
print('=====写入成功=====')
else:
print('=====写入失败=====\n', cursor.error)
#===========从数据库获取数据===========
select_sql = 'select * from models where id=1'
if cursor.execute(select_sql):
res = cursor.fetchall()
# 对读取的二进制进行处理
text = dealDBFileData(res)
# 将二进制写入新文件
with open(outDoc, 'wb') as f2:
f2.write(text)
f2.close()
print('=====读取成功=====')
cursor.close()
conn.close()