File: mappers.py

package info (click to toggle)
openstack-trove 1%3A24.0.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 6,976 kB
  • sloc: python: 50,665; sh: 2,866; makefile: 71
file content (118 lines) | stat: -rw-r--r-- 6,218 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
# Copyright 2011 OpenStack Foundation
# All Rights Reserved.
#
#    Licensed under the Apache License, Version 2.0 (the "License"); you may
#    not use this file except in compliance with the License. You may obtain
#    a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
#    License for the specific language governing permissions and limitations
#    under the License.

from sqlalchemy import MetaData
from sqlalchemy import orm
from sqlalchemy.orm import exc as orm_exc
from sqlalchemy.orm import registry
from sqlalchemy import Table

mapper_registry = registry()


def map(engine, models):
    meta = MetaData()
    meta.bind = engine
    if mapping_exists(models['instances']):
        return

    mapper_registry.map_imperatively(models['instances'],
                                     Table('instances', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(models['instance_faults'],
                                     Table('instance_faults', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(models['root_enabled_history'],
                                     Table('root_enabled_history', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(models['datastores'],
                                     Table('datastores', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(models['datastore_versions'],
                                     Table('datastore_versions', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(models['datastore_version_metadata'],
                                     Table('datastore_version_metadata',
                                           meta, autoload_with=engine))
    mapper_registry.map_imperatively(models['capabilities'],
                                     Table('capabilities', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(models['capability_overrides'],
                                     Table('capability_overrides', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(models['service_statuses'],
                                     Table('service_statuses', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(models['dns_records'],
                                     Table('dns_records', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(models['agent_heartbeats'],
                                     Table('agent_heartbeats', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(models['quotas'],
                                     Table('quotas', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(models['quota_usages'],
                                     Table('quota_usages', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(models['reservations'],
                                     Table('reservations', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(models['backups'],
                                     Table('backups', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(models['backup_strategy'],
                                     Table('backup_strategy', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(models['security_groups'],
                                     Table('security_groups', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(models['security_group_rules'],
                                     Table('security_group_rules', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(
        models['security_group_instance_associations'],
        Table('security_group_instance_associations', meta,
              autoload_with=engine))
    mapper_registry.map_imperatively(models['configurations'],
                                     Table('configurations', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(models['configuration_parameters'],
                                     Table('configuration_parameters',
                                           meta, autoload_with=engine))
    mapper_registry.map_imperatively(models['conductor_lastseen'],
                                     Table('conductor_lastseen', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(models['clusters'],
                                     Table('clusters', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(
        models['datastore_configuration_parameters'],
        Table('datastore_configuration_parameters', meta,
              autoload_with=engine))
    mapper_registry.map_imperatively(models['modules'],
                                     Table('modules', meta,
                                           autoload_with=engine))
    mapper_registry.map_imperatively(models['instance_modules'],
                                     Table('instance_modules', meta,
                                           autoload_with=engine))


def mapping_exists(model):
    try:
        orm.class_mapper(model)
        return True
    except orm_exc.UnmappedClassError:
        return False