python 實現資料庫中資料新增、查詢與更新的示例程式碼

阿新 • • 發佈：2020-12-09

一、前言

　　最近做web網站的測試，遇到很多需要批量造資料的功能；比如某個頁面展示資料條數需要達到10000條進行測試，此時手動構造資料肯定是不可能的，此時只能通過python指令碼進行自動構造資料；本次構造資料主要涉及到在某個表裡面批量新增資料、在關聯的幾個表中同步批量新增資料、批量查詢某個表中符合條件的資料、批量更新某個表中符合條件的資料等。　　

二、資料新增

　　即批量新增資料到某個表中。

insert_data.py

import pymysql
import random
import time
from get_userinfo import get_userinfo
from get_info import get_info
from get_tags import get_tags
from get_tuser_id import get_utag


class DatabaseAccess():
  def __init__(self):
    self.__db_host = "xxxxx"
    self.__db_port = 3307
    self.__db_user = "root"
    self.__db_password = "123456"
    self.__db_database = "xxxxxx"
  # 連線資料庫
  def isConnectionOpen(self):
    self.__db = pymysql.connect(
      host=self.__db_host,port=self.__db_port,user=self.__db_user,password=self.__db_password,database=self.__db_database,charset='utf8'
    )
  
  # 插入資料
  def linesinsert(self,n,user_id,tags_id,created_at):
 
    self.isConnectionOpen()
    # 建立遊標
    global cursor
    conn = self.__db.cursor()
    try:
      sql1 = '''
      INSERT INTO `codeforge_new`.`cf_user_tag`(`id`,`user_id`,`tag_id`,`created_at`,`updated_at`) VALUES ({},{},'{}','{}');
      '''.format(n,created_at,created_at)
      
      # 執行SQL  
      conn.execute(sql1,)
    except Exception as e:
      print(e)
    finally:
      # 關閉遊標
      conn.close()
      self.__db.commit()
      
      self.__db.close()
  
  def get_data(self):
    
    # 生成對應資料 1000條
    for i in range(0,1001):
      created_at = time.strftime('%Y-%m-%d %H:%M:%S',time.localtime())
      # print(create_at)
      # 使用者id
      tuserids = []
      tuserid_list = get_utag()
      for tuserid in tuserid_list:
        tuserids.append(tuserid[0])
      # print(tuserids)
      userid_list = get_userinfo()
      user_id = random.choice(userid_list)[0]
      if user_id not in tuserids:
        user_id=user_id
      
        # 標籤id
        tagsid_list = get_tags()
        tags_id = random.choice(tagsid_list)[0]
        self.linesinsert(i,created_at)


if __name__ == "__main__":
  # 例項化物件
  db=DatabaseAccess()
  db.get_data()

二、資料批量查詢

select_data.py

import pymysql
import pandas as pd
import numpy as np


def get_tags():
  # 連線資料庫，地址，埠，使用者名稱，密碼，資料庫名稱，資料格式
  conn = pymysql.connect(host='xxx.xxx.xxx.xxx',port=3307,user='root',passwd='123456',db='xxxx',charset='utf8')
  cur = conn.cursor()
  # 表cf_users中獲取所有使用者id
  sql = 'select id from cf_tags where id between 204 and 298'
  # 將user_id列轉成列表輸出
  df = pd.read_sql(sql,con=conn)
  # 先使用array()將DataFrame轉換一下
  df1 = np.array(df)
  # 再將轉換後的資料用tolist()轉成列表
  df2 = df1.tolist()
  # cur.execute(sql)
  # data = cur.fetchone()
  # print(df)
  # print(df1)
  # print(df2)
  return df2
  conn.close()

三、批量更新資料

select_data.py

import pymysql
import pandas as pd
import numpy as np


def get_tags():
  # 連線資料庫，地址，埠，使用者名稱，密碼，資料庫名稱，資料格式
  conn = pymysql.connect(host='xxx.xxx.xxx.xxx',con=conn)
  # 先使用array()將DataFrame轉換一下
  df1 = np.array(df)
  # 再將轉換後的資料用tolist()轉成列表
  df2 = df1.tolist()
  # cur.execute(sql)
  # data = cur.fetchone()
  # print(df)
  # print(df1)
  # print(df2)
  return df2
  conn.close()

以上就是python 實現資料庫中資料新增、查詢與更新的示例程式碼的詳細內容，更多關於python 資料庫新增、查詢與更新的資料請關注我們其它相關文章！