media limit

This commit is contained in:
2019-08-19 22:05:45 +03:00
parent 029d08d2aa
commit 8d564fb31b
2 changed files with 16 additions and 0 deletions

5
db.py
View File

@@ -1,3 +1,5 @@
from datetime import datetime
from BTrees.IIBTree import IIBTree
from ZODB import DB
from ZODB.Connection import Connection
@@ -21,6 +23,9 @@ def get_conn(read_only=False) -> Connection:
if not hasattr(user, 'messages_forward') or not isinstance(user.messages_forward, IIBTree):
user.messages_forward = IIBTree()
user.messages_reverse = IIBTree()
# migration 2
if not hasattr(conn.root, 'last_media'):
conn.root.last_media = datetime.now()
# end migrations
commit()
return conn