如何创建一个基于输入文件限制方法的类？创建数据库 API

Question

这个问题更多的是关于如何使用 OOP 读取数据库。 SQLite 和 sqlite3 只是可以使用的示例，并不是问题的主要内容。 （可以说Python在这里只是用来解释我的观点）：

我正在创建一个软件包，它允许用户查询已经生成的 SQLite 索引文件。对于非常特殊的情况，查询以某种方式索引的 SQLite 文件基本上是一种语法，这应该很简单，但我有点困惑如何“自动”读取 SQLite

这是一个示例（带有伪代码）：

import sqlite3

Class EasySQL:
    def __init__(self, filepath):
        self.filepath = filepath

    def connect(self, filepath):  ## perhaps this should be above in init?
        return sqlite3.connect(self.filepath)

    def query_indexA(self):
        ## query index A on SQLite3 connection

    def query_indexB(self):
        ## query index B on SQLite3 connection

    def query_indexC(self):
        ## query index C on SQLite3 connection

    def query_indexD(self):
        ## query index D on SQLite3 connection

这是我的问题：用户只能访问两种类型的索引 SQLite 文件。 “Type 1”文件仅在 A 和 B 上建立索引——索引 C 和 D 在此 SQLite 文件中不存在，如果用户尝试调用这些函数，应该会抛出错误。 “类型 2”文件仅在 C 和 D 上建立索引——该 SQLite 文件中不存在索引 A 和 B，如果用户尝试调用这些函数，应该会抛出错误。

我如何“限制”这些方法，以便如果用户在“Type 1”SQLite 文件上调用（例如）query_indexC，就会出现错误？

我强烈不希望强迫用户使用两个单独的类，因为这太复杂了，例如EasySQL_Type1() 或 EasySQL_Type2() 一个标志，但哪种类型可以。

Answer 1

如果您可以正确识别文件类型，您可以轻松地将其存储为内部引用，并在尝试访问“无效”索引时抛出错误。您还可以在__init__() 中建立与您的 SQLite 文件的连接，因为没有网络驱动器，您不需要重新连接处理。

当然，也有边缘情况，但这应该可以帮助您入门：

import sqlite3

class EasySQL(object):

    def __init__(self, filepath, index_type=1):  # type 1 by default
        self.connection = sqlite3.connect(filepath)
        self.index_type = index_type

    def query_indexA(self):
        if self.index_type != 1:
            raise ValueError("indexA can only be queried by index type 1")
        # do your indexA processing... your DB is available in self.connection

    def query_indexB(self):
        if self.index_type != 1:
            raise ValueError("indexB can only be queried by index type 1")
        # do your indexB processing... your DB is available in self.connection

    def query_indexC(self):
        if self.index_type != 2:
            raise ValueError("indexC can only be queried by index type 2")
        # do your indexC processing... your DB is available in self.connection

    def query_indexD(self):
        if self.index_type != 2:
            raise ValueError("indexD can only be queried by index type 2")
        # do your indexD processing... your DB is available in self.connection

现在您的用户可以通过以下方式启动它：

type1_db = EasySQL("path/to/their/file.sqlite", 1)  # for type 1, or
type2_db = EasySQL("path/to/their/file.sqlite", 2)  # for type 2

如果他们尝试使用与其类型无关的方法，则会引发ValueError。如果您不想让用户提供他们的数据库文件的类型，则必须设计一种从文件本身（名称、架构等）来确定它的方法。