python操作mysql、excel、pdf的示例

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

# 1. 定义Person类
class Person:

  def __init__(self, name, age):
    self.name = name
    self.age = age

  def watch_tv(self):
    print(f'{self.name} 看电视')


# 2. 定义loop函数
# 打印 1-max 中的奇数
def test_person():
  person = Person('Jake', 20)
  print(f'打印person的地址：', person)
  print(f'person.name:{person.name}')
  print(f'person.age:{person.age}')
  person.watch_tv()

  person = Person('Koko', 18)
  print(f'打印person的地址：', person)
  print(f'person.name:{person.name}')
  print(f'person.age:{person.age}')
  person.watch_tv()


# 3. 执行calculate方法
# 计算 当前值小于1，当前值：0
# 计算 1 >= 1: True
# 计算 2 >= 1: True
# 计算 10 >= 1: True
test_person()

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

# pip3 install pymysql

import pymysql

from getpass import getpass

# from mysql.connector import connect, Error
#
host = 'xxxxxxx'
port = 3306
username = 'db_account_member'
password = 'db_account_password'
database = 'some_database'


def connect_db():
  return pymysql.connect(host=host,
              port=port,
              user=username,
              password=password,
              database=database,
              charset='utf8')


def print_error(e):
  print(f'错误类型:{type(e)}')
  print(f'错误内容:{e}')


def close_gracefully(cursor, conn):
  if cursor:
    cursor.close()
  if conn:
    conn.close()


# 查询数据库，可以写任意查询语句
def query(sql):
  try:
    conn = connect_db() # 创建连接
    cursor = conn.cursor() # 建立游标
    cursor.execute(sql) # 执行sql语句
    return cursor.fetchall()
  except pymysql.Error as e:
    print_error(e)
  finally:
    close_gracefully(cursor, conn)



query_sql = 'select * from category where id = 1'
rows = query(query_sql)
print('category表中的数据如下:')
print(rows)

# -*- coding: UTF-8 -*-

# 1. 导入csv库
import csv

file_name = '../resources/test.csv'

# 2. 定义headers和rows
headers = ['index', 'name', 'sex', 'height', 'year']

rows = [
  [1, 'Jake', 'male', 177, 20],
  [2, 'Koko', 'female', 165, 18],
  [3, 'Mother', 'female', 163, 45],
  [4, 'Father', 'male', 172, 48]
]


# 3. 定义write_csv函数
# 写入csv
def write_csv():
  print(f'文件[{file_name}]准备写入')
  with open(f'{file_name}', 'w')as f:
    f_csv = csv.writer(f)
    f_csv.writerow(headers)
    f_csv.writerows(rows)
    print(f'文件[{file_name}]写入完毕')


# 读取csv
def read_csv():
  print(f'文件[{file_name}]准备读取')
  with open(f'{file_name}')as f:
    f_csv = csv.reader(f)
    for row in f_csv:
      print(row)
  print(f'文件[{file_name}]读取完毕')


# 4. 执行write_csv函数
write_csv()
print('------')
read_csv()

# -*- coding: UTF-8 -*-

# 导引
# 安装相关依赖
# pip3 install xlrd

# 引入xlrd去支持读取xls相关的文件
import xlrd

# 定义文件名
file_name = '../resources/sku.xls'

# 1. 读取xls文件
# 预计输出
# sku.xls该文档有 3 个tab页
sku_file = xlrd.open_workbook(file_name)
print("{0}该文档有 {1} 个tab页".format(file_name, sku_file.nsheets))
print("每个tab页，页名分别为: {0}".format(sku_file.sheet_names()))

# 2. 读取xls文件第1页
# 预计输出
# tab页名：Sheet1，该tab页共有59行，3列
# A6方格的值：1908165140370878
current_sheet_index = 0 # 下标0为第一页tab
current_sheet = sku_file.sheet_by_index(current_sheet_index)
print("tab页名：{0}，该tab页共有{1}行，{2}列".format(current_sheet.name, current_sheet.nrows, current_sheet.ncols))
print("A6方格的值：{0}".format(current_sheet.cell_value(rowx=5, colx=0)))

# 3. 打印每页的数据，每一行的数据为一个数组
# 预计输出
# [text:'1908154975415329', text:'鞋面是织物 鞋底是聚氨酯底的哦', text:'鞋底是5厘米 内增是3厘米 总高度是8厘米左右哦']
# [text:'1908040228021948', text:'鞋面是飞织 鞋底是聚氨酯底的哦', text:'鞋底高度是3厘米左右哦']
# ...以下省略后续打印
for rx in range(current_sheet.nrows):
  print(current_sheet.row(rx))

import platform
import pdfkit

# 这里根据自己的系统修改对应的wkhtmltopdf安装路径，修改其中一个就行了
win_path = 'D:/tools/wkhtmltopdf'
non_win_path = '/usr/local/bin/wkhtmltopdf'


def wkhtmltopdf_path():
  system = platform.system()
  if system == 'Darwin':
    print('苹果系统，可以生成pdf')
    path = non_win_path
  elif system == 'Windows':
    print('Windows系统，可以生成pdf')
    path = win_path
  elif system == 'Linux系统':
    print('Linux系统，可以生成pdf')
    path = non_win_path
  else:
    print('其他系统，暂不支持生成pdf')
    raise Exception('其他系统，暂不支持生成pdf')
  return path


def pre_config():
  return pdfkit.configuration(wkhtmltopdf=wkhtmltopdf_path())


# 从链接地址生成pdf
def generate_pdf_from_url(url, output_file_path):
  config = pre_config()
  pdfkit.from_url(url, output_file_path)


# 从字符串生成pdf
def generate_pdf_from_string(str, output_file_path):
  config = pre_config()
  pdfkit.from_string(str, output_file_path)


generate_pdf_from_url('https://baidu.com', '../temp/baidu_test.pdf')

generate_pdf_from_string('hello', '../temp/hello.pdf')