用Python编写一个简单的数据库管理工具

Python是一种简单易用的编程语言，可以用来编写各种类型的应用程序，包括数据库管理工具。数据库管理工具可以帮助用户连接、查询、修改和管理数据库中的数据。下面是一个使用Python编写的简单数据库管理工具的例子。

import sqlite3

def create_connection():
    conn = None;
    try:
        conn = sqlite3.connect('test.db')
        print('Database connection established.')
    except sqlite3.Error as e:
        print(e)
        
    if conn:
        conn.close()
        print('Database connection closed.')
        
def create_table():
    conn = sqlite3.connect('test.db')
    c = conn.cursor()
    try:
        c.execute('''CREATE TABLE IF NOT EXISTS users
                     (id INTEGER PRIMARY KEY AUTOINCREMENT,
                      name TEXT NOT NULL,
                      age INTEGER NOT NULL);''')
        print('Table created successfully.')
    except sqlite3.Error as e:
        print(e)
        
    conn.commit()
    conn.close()
    
def insert_data():
    conn = sqlite3.connect('test.db')
    c = conn.cursor()
    
    name = input('Enter name: ')
    age = int(input('Enter age: '))
    
    try:
        c.execute("INSERT INTO users(name, age) VALUES (?, ?)", (name, age))
        print('Data inserted successfully.')
    except sqlite3.Error as e:
        print(e)
        
    conn.commit()
    conn.close()
    
def display_data():
    conn = sqlite3.connect('test.db')
    c = conn.cursor()
    
    try:
        c.execute("SELECT * FROM users")
        rows = c.fetchall()
        for row in rows:
            print('ID: {}, Name: {}, Age: {}'.format(row[0], row[1], row[2]))
    except sqlite3.Error as e:
        print(e)
        
    conn.commit()
    conn.close()
    
if __name__ == '__main__':
    create_connection()
    create_table()
    
    while True:
        print('1. Insert Data')
        print('2. Display Data')
        print('3. Exit')
        
        choice = int(input('Enter your choice: '))
        
        if choice == 1:
            insert_data()
        elif choice == 2:
            display_data()
        elif choice == 3:
            break
        else:
            print('Invalid choice. Please try again.')

这个简单的数据库管理工具使用了SQLite数据库，涵盖了创建数据库连接、创建表格、向表格插入数据、显示数据等功能。使用者可以通过命令行菜单选择不同的操作。

首先，create_connection函数会尝试连接到test.db数据库文件。如果连接成功，会打印数据库连接成功的信息，然后关闭连接。

然后，create_table函数会创建一个名为users的表格，包括id、name和age三个列。如果表格已经存在，就会提示表格已经存在的信息。创建表格成功后会提交更改并关闭连接。

接下来，insert_data函数会通过用户的输入向users表格插入数据。用户需要输入一个姓名和年龄，然后数据会插入到表格中。插入数据成功后会提交更改并关闭连接。

最后，display_data函数会将users表格中的所有数据显示出来。查询数据成功后会提交查询并关闭连接。

在__main__函数中，会先调用create_connection和create_table函数创建数据库和表格。然后进入一个循环，用户可以选择插入数据、显示数据或者退出程序。

以上是一个简单的使用Python编写的数据库管理工具的例子，其中涵盖了连接数据库、创建表格、插入数据和显示数据等功能。你可以根据自己的需求进行修改和扩展。