1 from sqlalchemy import *
2 from sqlalchemy import orm
3 from sqlalchemy.orm import create_session, relation
7 from invirt.database import record
9 # Compatibility hack from
10 # http://www.sqlalchemy.org/trac/wiki/UsageRecipes/SessionAwareMapper
11 from sqlalchemy.orm import mapper as sqla_mapper
13 def session_mapper(scoped_session):
14 def mapper(cls, *arg, **kw):
15 if cls.__init__ is object.__init__:
16 def __init__(self, **kwargs):
17 for key, value in kwargs.items():
18 setattr(self, key, value)
19 scoped_session.add(self)
20 cls.__init__ = __init__
21 cls.query = scoped_session.query_property()
22 return sqla_mapper(cls, *arg, **kw)
30 'machine_access_table',
54 meta = ThreadLocalMetaData()
55 session = orm.scoped_session(orm.sessionmaker(autocommit=True, autoflush=False))
57 machine_table = Table('machines', meta,
58 Column('machine_id', Integer, primary_key=True, nullable=False),
59 Column('name', String, nullable=False, unique=True),
60 Column('description', String, nullable=False),
61 Column('memory', Integer, nullable=False),
62 Column('owner', String, nullable=False),
63 Column('contact', String, nullable=False),
64 Column('uuid', String, nullable=False),
65 Column('administrator', String, nullable=True, default=None),
66 Column('type_id', String, ForeignKey('types.type_id'), nullable=False),
67 Column('autorestart', Boolean, nullable=False, default=False),
68 Column('cpus', Integer, nullable=False, default=1),
69 Column('adminable', Boolean, nullable=False, default=False))
71 nic_table = Table('nics', meta,
72 Column('machine_id', Integer, ForeignKey('machines.machine_id'), nullable=True),
73 Column('mac_addr', String, nullable=False, primary_key=True),
74 Column('nic_type', String, nullable=False, default="pcnet"),
75 Column('ip', String, nullable=False, unique=True),
76 Column('netmask', String, nullable=False),
77 Column('gateway', String, nullable=False),
78 Column('other_ip', String, nullable=True),
79 Column('other_netmask', String, nullable=True),
80 Column('other_gateway', String, nullable=True),
81 Column('other_action', String, nullable=True),
82 Column('hostname', String, nullable=True),
83 Column('reusable', Boolean, nullable=False, default=True))
85 disk_table = Table('disks', meta,
86 Column('machine_id', Integer, ForeignKey('machines.machine_id'), nullable=False),
87 Column('guest_device_name', String, nullable=False),
88 Column('size', Integer, nullable=False),
89 PrimaryKeyConstraint('machine_id', 'guest_device_name'))
91 types_table = Table('types', meta,
92 Column('type_id', String, primary_key=True, nullable=False),
93 Column('description', String, nullable=False),
94 Column('hvm', Boolean, nullable=False),
95 Column('apic', Boolean, nullable=False),
96 Column('acpi', Boolean, nullable=False),
97 Column('pae', Boolean, nullable=False))
99 mirrors_table = Table('mirrors', meta,
100 Column('mirror_id', String, primary_key=True, nullable=False),
101 Column('uri_prefix', String, nullable=False))
103 cdroms_table = Table('cdroms', meta,
104 Column('cdrom_id', String, primary_key=True, nullable=False),
105 Column('description', String, nullable=False),
106 Column('mirror_id', String, ForeignKey('mirrors.mirror_id')),
107 Column('uri_suffix', String))
109 autoinstalls_table = Table('autoinstalls', meta,
110 Column('autoinstall_id', String, primary_key=True, nullable=False),
111 Column('description', String, nullable=False),
112 Column('type_id', String, ForeignKey('types.type_id'), nullable=False),
113 Column('distribution', String, nullable=False),
114 Column('mirror', String, nullable=False),
115 Column('arch', String, nullable=False),
116 Column('preseed', Boolean, nullable=False, default=False))
118 owners_table = Table('owners', meta,
119 Column('owner_id', String, primary_key=True, nullable=False),
120 Column('ram_quota_total', Integer, nullable=True, default=None),
121 Column('ram_quota_single', Integer, nullable=True, default=None),
122 Column('disk_quota_total', Integer, nullable=True, default=None),
123 Column('disk_quota_single', Integer, nullable=True, default=None),
124 Column('vms_quota_total', Integer, nullable=True, default=None),
125 Column('vms_quota_active', Integer, nullable=True, default=None))
127 builds_table = Table('builds', meta,
128 Column('build_id', Integer, primary_key=True, nullable=False),
129 Column('package', String, nullable=False),
130 Column('pocket', String, nullable=False),
131 Column('commit', String, nullable=False),
132 Column('version', String, nullable=True, default=None),
133 Column('principal', String, nullable=False),
134 Column('succeeded', Boolean, nullable=True),
135 Column('failed_stage', String, nullable=True, default=None),
136 Column('traceback', String, nullable=True, default=None),
137 Column('inserted_at', DateTime, nullable=False, default=datetime.datetime.utcnow))
139 machine_access_table = Table('machine_access', meta,
140 Column('machine_id', Integer, ForeignKey('machines.machine_id', ondelete='CASCADE'), nullable=False, index=True),
141 Column('user', String, nullable=False, index=True),
142 PrimaryKeyConstraint('machine_id', 'user'))
144 admins_table = Table('admins', meta,
145 Column('user', String, nullable=False, index=True, primary_key=True))
147 class Machine(record.Record):
148 _identity_field = 'name'
150 class MachineAccess(record.Record):
153 class NIC(record.Record):
156 class Disk(record.Record):
159 class Type(record.Record):
160 _identity_field = 'type_id'
162 class Mirror(record.Record):
163 _identity_field = 'mirror_id'
165 class CDROM(record.Record):
166 _identity_field = 'cdrom_id'
168 class Autoinstall(record.Record):
169 _identity_field = 'autoinstall_id'
171 class Admin(record.Record):
172 _identity_field = 'user'
174 class Build(record.Record):
175 _identity_field = 'build_id'
177 from invirt.database.owner import Owner
178 mapper = session_mapper(session)
180 mapper(Machine, machine_table,
181 properties={'nics': relation(NIC, backref="machine"),
182 'disks': relation(Disk, backref="machine"),
183 'type': relation(Type),
184 'acl': relation(MachineAccess, backref="machine", passive_deletes=True, cascade="all, delete-orphan")});
185 mapper(MachineAccess, machine_access_table)
186 mapper(NIC, nic_table)
187 mapper(Disk, disk_table)
188 mapper(Type, types_table)
189 mapper(Mirror, mirrors_table)
190 mapper(CDROM, cdroms_table,
191 properties={'mirror': relation(Mirror, backref="cdroms")})
192 mapper(Autoinstall, autoinstalls_table)
193 mapper(Owner, owners_table)
194 mapper(Admin, admins_table)
195 mapper(Build, builds_table)
198 """Clear sqlalchemy's cache
201 session.expunge_all()