# -*- coding: utf-8 -*-

from puremysql import PureMysql

db_config = {
    "database": "mydata",
    "user": "root",
    "password": "123456",
    "host": "127.0.0.1",
    "port": 3306,
}

pure_mysql = PureMysql(**db_config)

# 或者
# url = "mysql://root:123456@127.0.0.1:3306/mydata"
# pure_mysql = PureMysql(db_url=url)

student = pure_mysql.table("student")


# 插入数据
def test_insert():
    data = {
        "name": "Tom",
        "age": 25
    }

    count = student.insert(data)
    print(count)  # 1

    data = [
        {
            "name": "Tom",
            "age": 26,

        },
        {
            "name": "Jack",
            "age": 27,

        }
    ]
    count = student.insert(data)
    print(count)  # 2


# 删除数据
def test_delete():
    ret = student.delete("id=13")
    print(ret)  # 1

    ret = student.delete_by_id(12)
    print(ret)  # 1


# 更新数据
def test_update():
    data = {
        "name": "Tom"
    }
    ret = student.update(data, "name='Tom'")
    print(ret)  # 1

    ret = student.update_by_id(data, 1)
    print(ret)  # 1


# 查询数据
def test_select():
    ret = student.select(["name", "age"], "id=1")
    print(ret)
    # [{"name": "Tom", "age": 25}]

    ret = student.select("name, age", "id=1")
    print(ret)
    # [{"name": "Tom", "age": 25}]

    ret = student.select_one(["name", "age"], "id=1")
    print(ret)
    # {"name": "Tom", "age": 25}

    ret = student.select_by_id("name, age", 2007)
    print(ret)
    # {"name": "Tom", "age": 25}