mirror of https://github.com/jumpserver/jumpserver
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
|
# Generated by Django 4.1.13 on 2024-03-26 07:31
|
||
|
import json
|
||
|
|
||
|
from django.db import migrations
|
||
|
from django.db.models import F
|
||
|
|
||
|
|
||
|
def migrate_feishu_to_lark(apps, schema_editor):
|
||
|
setting_model = apps.get_model("settings", "Setting")
|
||
|
user_model = apps.get_model("users", "User")
|
||
|
db_alias = schema_editor.connection.alias
|
||
|
|
||
|
feishu_version_instance = setting_model.objects.using(db_alias).filter(name='FEISHU_VERSION').first()
|
||
|
if not feishu_version_instance or json.loads(feishu_version_instance.value) == 'feishu':
|
||
|
return
|
||
|
feishu_version_instance.delete()
|
||
|
|
||
|
user_model.objects.using(db_alias).filter(feishu_id__isnull=False).update(lark_id=F('feishu_id'))
|
||
|
user_model.objects.filter(feishu_id__isnull=False).update(lark_id=F('feishu_id'))
|
||
|
user_model.objects.filter(feishu_id__isnull=False).update(feishu_id=None)
|
||
|
|
||
|
settings_to_update = [
|
||
|
('AUTH_FEISHU', 'AUTH_LARK'),
|
||
|
('FEISHU_APP_ID', 'LARK_APP_ID'),
|
||
|
('FEISHU_APP_SECRET', 'LARK_APP_SECRET'),
|
||
|
]
|
||
|
|
||
|
for old_name, new_name in settings_to_update:
|
||
|
setting_model.objects.using(db_alias).filter(
|
||
|
name=old_name
|
||
|
).update(name=new_name)
|
||
|
|
||
|
|
||
|
class Migration(migrations.Migration):
|
||
|
dependencies = [
|
||
|
('settings', '0012_alter_setting_options'),
|
||
|
('users', '0050_user_lark_id_alter_user_source'),
|
||
|
]
|
||
|
|
||
|
operations = [
|
||
|
migrations.RunPython(migrate_feishu_to_lark),
|
||
|
]
|