Lib MySQLdb.py

1 개요

lib_MySQLdb.py

import os
import MySQLdb
db = MySQLdb.connect(
  host = os.getenv('DB_HOST'),
  port = os.getenv('DB_PORT'),
  user = os.getenv('DB_USER'),
  passwd = os.getenv('DB_PASS'),
  db = os.getenv('DB_DATABASE')
)
db.autocommit(True)
mycursor = db.cursor()

def query(sql,params=()):
  mycursor.execute(sql,params)

def query_rows(sql,params=()):
  mycursor.execute(sql,params)
  columns = [col[0] for col in mycursor.description]
  return [dict(zip(columns, row)) for row in mycursor.fetchall()]

def query_row(sql,params=()):
  rows = query_rows(sql,params)
  if len(rows) < 1:
    return False
  return rows[0]

def query_one(sql,params=()):
  mycursor.execute(sql,params)
  rows = mycursor.fetchall()
  return rows[0][0]

2 사용예시

tables = query_rows("SHOW TABLES")
print(tables)

3 같이 보기

• lib_mysql_connector.py