def database(self, db_name): """ Params: db_name (str): name of a database Returns: :class:`tdclient.models.Database` """ databases = self.api.list_databases() for (name, kwargs) in databases.items(): if name == db_name: return models.Database(self, name, **kwargs) raise api.NotFoundError("Database '%s' does not exist" % (db_name))
def table(self, db_name, table_name): """ Args: db_name (str): name of a database table_name (str): name of a table Returns: :class:`tdclient.models.Table` Raises: tdclient.api.NotFoundError: if the table doesn't exist """ tables = self.tables(db_name) for table in tables: if table.table_name == table_name: return table raise api.NotFoundError("Table '%s.%s' does not exist" % (db_name, table_name))