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/rds2/models.py
View file

@ -7,7 +7,7 @@ import boto.rds2
from jinja2 import Template
from re import compile as re_compile
from moto.cloudformation.exceptions import UnformattedGetAttTemplateException
from moto.core import BaseBackend
from moto.core import BaseBackend, BaseModel
from moto.core.utils import get_random_hex
from moto.ec2.models import ec2_backends
from .exceptions import (RDSClientError,
@ -17,7 +17,7 @@ from .exceptions import (RDSClientError,
DBParameterGroupNotFoundError)
class Database(object):
class Database(BaseModel):
def __init__(self, **kwargs):
self.status = "available"
@ -372,7 +372,7 @@ class Database(object):
backend.delete_database(self.db_instance_identifier)
class SecurityGroup(object):
class SecurityGroup(BaseModel):
def __init__(self, group_name, description, tags):
self.group_name = group_name
@ -481,7 +481,7 @@ class SecurityGroup(object):
backend.delete_security_group(self.group_name)
class SubnetGroup(object):
class SubnetGroup(BaseModel):
def __init__(self, subnet_name, description, subnets, tags):
self.subnet_name = subnet_name