catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Add BaseModel to all models.

Browse source
This commit is contained in:
Steve Pulec 2017-03-11 23:41:12 -05:00
commit 6d422d1f37
34 changed files with 122 additions and 116 deletions
Download patch file Download diff file Expand all files Collapse all files

8
moto/ses/models.py
View file

@ -2,7 +2,7 @@ from __future__ import unicode_literals
import email
from moto.core import BaseBackend
from moto.core import BaseBackend, BaseModel
from .exceptions import MessageRejectedError
from .utils import get_random_message_id
@ -10,19 +10,19 @@ from .utils import get_random_message_id
RECIPIENT_LIMIT = 50
class Message(object):
class Message(BaseModel):
def __init__(self, message_id):
self.id = message_id
class RawMessage(object):
class RawMessage(BaseModel):
def __init__(self, message_id):
self.id = message_id
class SESQuota(object):
class SESQuota(BaseModel):
def __init__(self, sent):
self.sent = sent