bakatrouble/telegram_bots
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

hide mtg fields

Browse Source
This commit is contained in:
bakatrouble
2019-11-28 00:12:36 +03:00
parent 1f8ffb5cba
commit 890b8b4493
4 changed files with 59 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
bots/modules/cyberlina.py
View File

@@ -26,6 +26,7 @@ class CyberLinaBotModuleConfig(TelegramBotModuleConfig):
welcome_reactions = JSONField(default='[]')
inline_reactions = JSONField(default='[]')
_mtg_data = JSONField(default={}, blank=True, null=True)
_mtg_corpus = models.TextField(null=True, blank=True)
mtg_train = models.TextField(null=True, blank=True)
MODULE_NAME = 'Киберлиночка'
@@ -38,7 +39,7 @@ class CyberLinaBotModuleConfig(TelegramBotModuleConfig):
'welcome_reactions': JSONEditor(),
'inline_reactions': JSONEditor(),
}
EXCLUDE_FIELDS = '_mtg_data',
EXCLUDE_FIELDS = '_mtg_data', '_mtg_corpus',
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
@@ -120,7 +121,9 @@ class CyberLinaBotModuleConfig(TelegramBotModuleConfig):
def save(self, force_insert=False, force_update=False, using=None, update_fields=None):
if self.mtg_train:
self._mtg_data = markovify.Text(self.mtg_train).to_dict()
self._mtg_corpus += '\n' + self.mtg_train
self._mtg_data = markovify.Text(self._mtg_corpus).to_dict()
self.mtg_train = None
super().save(force_insert, force_update, using, update_fields)