Instantiate dataset.Database object once and return it without re-instantiating it.

This commit is contained in:
Davte 2019-06-05 12:49:47 +02:00
parent 0fae3d305a
commit 18fb69dff8
2 changed files with 15 additions and 5 deletions

View File

@ -7,7 +7,7 @@ __author__ = "Davide Testa"
__email__ = "davide@davte.it" __email__ = "davide@davte.it"
__credits__ = "Marco Origlia" __credits__ = "Marco Origlia"
__license__ = "GNU General Public License v3.0" __license__ = "GNU General Public License v3.0"
__version__ = "1.5.4" __version__ = "1.5.5"
__maintainer__ = "Davide Testa" __maintainer__ = "Davide Testa"
__contact__ = "t.me/davte" __contact__ = "t.me/davte"

View File

@ -166,8 +166,10 @@ class Bot(telepot.aio.Bot, Gettable):
name=db_name, name=db_name,
ext='.db' if not db_name.endswith('.db') else '' ext='.db' if not db_name.endswith('.db') else ''
) )
self._database = dataset.connect(self.db_url)
else: else:
self._db_url = None self._db_url = None
self._database = None
self._unauthorized_message = None self._unauthorized_message = None
self.authorization_function = lambda update, authorization_level: True self.authorization_function = lambda update, authorization_level: True
self.get_chat_id = lambda update: ( self.get_chat_id = lambda update: (
@ -222,12 +224,20 @@ class Bot(telepot.aio.Bot, Gettable):
@property @property
def db(self): def db(self):
"""Connect to bot's database. """Return the dataset.Database instance related to `self`.
It must be used inside a with statement: `with bot.db as db` To start a transaction with bot's database, use a with statement:
```python3
with bot.db as db:
db['your_table'].insert(
dict(
name='John',
age=45
)
)
```
""" """
if self.db_url: return self._database
return dataset.connect(self.db_url)
@property @property
def default_keyboard(self): def default_keyboard(self):