import pymysql


def connection():
    conn = pymysql.connect(db='x',
                       user='x',
                       passwd='x',
                       host='x')
    return conn


def test(conn):
    with conn.cursor() as cursor:
        sql = "SELECT id from xxxx"
        cursor.execute(sql)
        result = cursor.fetchall()
        print(result)

def main():
    test(connection())

main()