pymysql

import pymysql

host='localhost'
user='root'
password=''
database='anjuke'
cursorclass=pymysql.cursors.DictCursor

def select_count(sql):
        #连接数据库
        connection = pymysql.connect(host=host,
user=user,
password=password,
database=database,
cursorclass=cursorclass)

        try:
            with connection.cursor() as cursor:
                cursor.execute(sql)
                all = cursor.fetchall()
                connection.commit()
                return all[0]['count(*)']
        except:
            print('sql error')
        finally:
            # 关闭连接
            connection.close()


def execute(sql):
  # 连接数据库
  connection = pymysql.connect(host=host,
                               user=user,
                               password=password,
                               database=database,
                               cursorclass=cursorclass)

  try:
    with connection.cursor() as cursor:
      cursor.execute(sql)
      connection.commit()
  except:
    print('sql error')
  finally:
    # 关闭连接
    connection.close()