mirror of
https://github.com/valitydev/redash.git
synced 2024-11-07 01:25:16 +00:00
36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
|
from redash.models import db, Organization, Group
|
||
|
|
from redash import settings
|
||
|
|
from playhouse.migrate import PostgresqlMigrator, migrate
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
migrator = PostgresqlMigrator(db.database)
|
||
|
|
|
||
|
|
with db.database.transaction():
|
||
|
|
Organization.create_table()
|
||
|
|
|
||
|
|
default_org = Organization.create(name="Default", settings={
|
||
|
|
Organization.SETTING_GOOGLE_APPS_DOMAINS: settings.GOOGLE_APPS_DOMAIN
|
||
|
|
})
|
||
|
|
|
||
|
|
column = Group.org
|
||
|
|
column.default = default_org
|
||
|
|
|
||
|
|
migrate(
|
||
|
|
migrator.add_column('groups', 'org_id', column),
|
||
|
|
migrator.add_column('events', 'org_id', column),
|
||
|
|
migrator.add_column('data_sources', 'org_id', column),
|
||
|
|
migrator.add_column('users', 'org_id', column),
|
||
|
|
migrator.add_column('dashboards', 'org_id', column),
|
||
|
|
migrator.add_column('queries', 'org_id', column),
|
||
|
|
migrator.add_column('query_results', 'org_id', column),
|
||
|
|
)
|
||
|
|
|
||
|
|
# Change the uniqueness constraint on user email to be (org, email):
|
||
|
|
migrate(
|
||
|
|
migrator.drop_index('users', 'users_email'),
|
||
|
|
migrator.add_index('users', ('org_id', 'email'), unique=True)
|
||
|
|
)
|
||
|
|
|
||
|
|
db.close_db(None)
|
||
|
|
|