Ошибка в питоне возникла)

Question

Ошибка в питоне возникла)

Артём Чернышов Ученик (226), закрыт 6 месяцев назад

я пытаюсь импортировать класс DB из самописанного модуля Database.

 from Database import DB

но он выдает мне это:

 Traceback (most recent call last): 
  File "c:\Users\inkoa\Desktop\python-testings!\my_bot\main.py", line 5, in  
    db = Database.DB('database.db') 
         ^^^^^^^^^^^ 
AttributeError: module 'Database' has no attribute 'DB'

Код самого модуля выглядит вот так:

 import sqlite3 
 
class DB: 
    def __init__(self, database_file: str): 
        self.connection = sqlite3.connect(database_file) 
        self.cursor = self.connection.cursor() 
     
    def new_user(self, user_id: int): 
        with self.connection: 
            self.cursor.execute("INSERT INTO users ('user_id') VALUES(?,)", (user_id)) 
    def user_exist(self, user_id: int): 
        with self.connection: 
            result = self.cursor.execute("SELECT * FROM users WHERE 'user_id' = ?", (user_id,)) 
            return bool(len(result))

Почему он не видит класс DB?

Answer 1

Либо:

 import Database
db = Database.DB('database.db')

Либо:

 from Database import DB
db = DB('database.db')

Но никак не вперемешку.

Answer 2

import sqlite3

class DB:
def __init__(self, database_file: str):
self.connection = sqlite3.connect(database_file)
self.cursor = self.connection.cursor()

def new_user(self, user_id: int):
with self.connection:
self.cursor.execute("INSERT INTO users (user_id) VALUES(?)", (user_id,))

def user_exist(self, user_id: int):
with self.connection:
result = self.cursor.execute("SELECT * FROM users WHERE user_id = ?", (user_id,))
return bool(len(result.fetchall()))