We now need to import and use our database module inside our server module. In order to make this easier, we should first wrap all of our functions in a class.
Update your database.py
file, creating a class named Database
and adding the necessary self
instances to your methods. We can also move the database name out to an attribute in the __init__
method:
import sqlite3 class Database: def __init__(self): self.database = "chat.db" def perform_insert(self, sql, params): conn = sqlite3.connect(self.database) ... def perform_select(self, sql, params): conn = sqlite3.connect(self.database) ... # update the rest of your methods to include self where necessary
Now that we have that done, we can import and instantiate the database in our server.py
file:
... from database import Database ... database = Database() ...
With our database in place, we can go ahead and turn the get_all_users
method into an endpoint available...