Question

我目前正在修读网页设计课程。我想要做的是检查我的数据库中是否存在名为portafolio的表，如果不是，我想创建一个表。我正在使用Python（flask）和sqlite3来管理我的数据库。到目前为止，我在SQL中的一些逻辑创建了一个表，如果它不存在：

# db is my database variable
db.execute('''create table if not exists portafolio(id INTEGER PRIMARY KEY AUTOINCREMENT, 
                                              stock TEXT, 
                                              shares INTEGER, 
                                              price FLOAT(2), 
                                              date TEXT
                                             ''');

但是我不想使用SQL命令，而是想知道如何在Python中进行完全相同的检查，因为它看起来会更清晰。

任何帮助都将不胜感激。

Answer 1

不确定哪种方式更干净，但您可以发出简单的选择并处理异常：

try:
    cursor.execute("SELECT 1 FROM portafolio LIMIT 1;")
    exists = True
except sqlite3.OperationalError as e:
    message = e.args[0]
    if message.startswith("no such table"):
        print("Table 'portafolio' does not exist")
        exists = False
    else:
        raise

请注意，我们必须检查它是什么类型的OperationalError，并且我们必须在字符串检查中使用这个“不漂亮”的消息子字符串，因为there is currently no way to get the actual error code。

或者，更多SQLite specific approach：

table_name = "portafolio"
cursor.execute("""
    SELECT name 
    FROM sqlite_master 
    WHERE type='table' AND name=?;
""", (table_name, ))

exists = bool(cursor.fetchone())

Answer 2

如果您正在寻找数据库操作的整洁工作，我建议您了解有关ORM（对象关系模型）的更多信息。

我将Flask与SQLAlchemy一起使用。您可以使用python类来管理SQL操作，如下所示：

class Portafolio(db.Model):
   id = db.Column(db.Integer, primary_key=True)
   stock = db.Column(db.String(255), unique=True)
   shares = db.Column(db.Integer, unique=True)

它可以轻松完成所有数据库检查和迁移。

如何检查Python中是否存在表？

2 个答案: