Completed
Push — master ( ac93a9...1372d4 )
by Daniel
04:23 queued 02:25
created

_create_user_class()   A

Complexity

Conditions 1

Size

Total Lines 9

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
cc 1
c 0
b 0
f 0
dl 0
loc 9
rs 9.6666
1
from sqlalchemy import Column, Integer, String
2
3
from groundwork_web.patterns import GwWebDbAdminPattern
4
5
6
def _create_user_class(Base):
7
    class User(Base):
8
        __tablename__ = 'users'
9
        id = Column(Integer, primary_key=True)
10
        name = Column(String)
11
        fullname = Column(String)
12
        password = Column(String)
13
14
    return User
15
16
17
class WebDatabasePlugin(GwWebDbAdminPattern):
18
    def __init__(self, *args, **kwargs):
19
        self.name = kwargs.get("name", self.__class__.__name__)
20
        super().__init__(*args, **kwargs)
21
22
        my_db = self.databases.register("main", "sqlite://", "main test database")
23
        User = _create_user_class(my_db.Base)
24
        my_User = my_db.classes.register(User)
25
26
        self.web.db.register(my_User, my_db.session)
27
28
    def activate(self):
29
        pass
30
31
    def deactivate(self):
32
        pass
33