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 cls.__init__ = __init__
20 cls.query = scoped_session.query_property()
21 return sqla_mapper(cls, *arg, **kw)
29 'machine_access_table',
53 meta = ThreadLocalMetaData()
54 session = orm.scoped_session(orm.sessionmaker(autocommit=True, autoflush=False))
56 machine_table = Table('machines', meta,
57 Column('machine_id', Integer, primary_key=True, nullable=False),
58 Column('name', String, nullable=False, unique=True),
59 Column('description', String, nullable=False),
60 Column('memory', Integer, nullable=False),
61 Column('owner', String, nullable=False),
62 Column('contact', String, nullable=False),
63 Column('uuid', String, nullable=False),
64 Column('administrator', String, nullable=True, default=None),
65 Column('type_id', String, ForeignKey('types.type_id'), nullable=False),
66 Column('autorestart', Boolean, nullable=False, default=False),
67 Column('cpus', Integer, nullable=False, default=1),
68 Column('adminable', Boolean, nullable=False, default=False))
70 nic_table = Table('nics', meta,
71 Column('machine_id', Integer, ForeignKey('machines.machine_id'), nullable=True),
72 Column('mac_addr', String, nullable=False, primary_key=True),
73 Column('ip', String, nullable=False, unique=True),
74 Column('hostname', String, nullable=True),
75 Column('reusable', Boolean, nullable=False, default=True))
77 disk_table = Table('disks', meta,
78 Column('machine_id', Integer, ForeignKey('machines.machine_id'), nullable=False),
79 Column('guest_device_name', String, nullable=False),
80 Column('size', Integer, nullable=False),
81 PrimaryKeyConstraint('machine_id', 'guest_device_name'))
83 types_table = Table('types', meta,
84 Column('type_id', String, primary_key=True, nullable=False),
85 Column('description', String, nullable=False),
86 Column('hvm', Boolean, nullable=False),
87 Column('apic', Boolean, nullable=False),
88 Column('acpi', Boolean, nullable=False),
89 Column('pae', Boolean, nullable=False))
91 mirrors_table = Table('mirrors', meta,
92 Column('mirror_id', String, primary_key=True, nullable=False),
93 Column('uri_prefix', String, nullable=False))
95 cdroms_table = Table('cdroms', meta,
96 Column('cdrom_id', String, primary_key=True, nullable=False),
97 Column('description', String, nullable=False),
98 Column('mirror_id', String, ForeignKey('mirrors.mirror_id')),
99 Column('uri_suffix', String))
101 autoinstalls_table = Table('autoinstalls', meta,
102 Column('autoinstall_id', String, primary_key=True, nullable=False),
103 Column('description', String, nullable=False),
104 Column('type_id', String, ForeignKey('types.type_id'), nullable=False),
105 Column('distribution', String, nullable=False),
106 Column('mirror', String, nullable=False),
107 Column('arch', String, nullable=False),
108 Column('preseed', Boolean, nullable=False, default=False))
110 owners_table = Table('owners', meta,
111 Column('owner_id', String, primary_key=True, nullable=False),
112 Column('ram_quota_total', Integer, nullable=True, default=None),
113 Column('ram_quota_single', Integer, nullable=True, default=None),
114 Column('disk_quota_total', Integer, nullable=True, default=None),
115 Column('disk_quota_single', Integer, nullable=True, default=None),
116 Column('vms_quota_total', Integer, nullable=True, default=None),
117 Column('vms_quota_active', Integer, nullable=True, default=None))
119 builds_table = Table('builds', meta,
120 Column('build_id', Integer, primary_key=True, nullable=False),
121 Column('package', String, nullable=False),
122 Column('pocket', String, nullable=False),
123 Column('commit', String, nullable=False),
124 Column('version', String, nullable=True, default=None),
125 Column('principal', String, nullable=False),
126 Column('succeeded', Boolean, nullable=True),
127 Column('failed_stage', String, nullable=True, default=None),
128 Column('traceback', String, nullable=True, default=None),
129 Column('inserted_at', DateTime, nullable=False, default=datetime.datetime.utcnow))
131 machine_access_table = Table('machine_access', meta,
132 Column('machine_id', Integer, ForeignKey('machines.machine_id', ondelete='CASCADE'), nullable=False, index=True),
133 Column('user', String, nullable=False, index=True),
134 PrimaryKeyConstraint('machine_id', 'user'))
136 admins_table = Table('admins', meta,
137 Column('user', String, nullable=False, index=True, primary_key=True))
139 class Machine(record.Record):
140 _identity_field = 'name'
142 class MachineAccess(record.Record):
145 class NIC(record.Record):
148 class Disk(record.Record):
151 class Type(record.Record):
152 _identity_field = 'type_id'
154 class Mirror(record.Record):
155 _identity_field = 'mirror_id'
157 class CDROM(record.Record):
158 _identity_field = 'cdrom_id'
160 class Autoinstall(record.Record):
161 _identity_field = 'autoinstall_id'
163 class Admin(record.Record):
164 _identity_field = 'user'
166 class Build(record.Record):
167 _identity_field = 'build_id'
169 from invirt.database.owner import Owner
170 mapper = session_mapper(session)
172 mapper(Machine, machine_table,
173 properties={'nics': relation(NIC, backref="machine"),
174 'disks': relation(Disk, backref="machine"),
175 'type': relation(Type),
176 'acl': relation(MachineAccess, backref="machine", passive_deletes=True, cascade="all, delete-orphan")});
177 mapper(MachineAccess, machine_access_table)
178 mapper(NIC, nic_table)
179 mapper(Disk, disk_table)
180 mapper(Type, types_table)
181 mapper(Mirror, mirrors_table)
182 mapper(CDROM, cdroms_table,
183 properties={'mirror': relation(Mirror, backref="cdroms")})
184 mapper(Autoinstall, autoinstalls_table)
185 mapper(Owner, owners_table)
186 mapper(Admin, admins_table)
187 mapper(Build, builds_table)
190 """Clear sqlalchemy's cache