改密计划支持数据库(mysql、oracle、postgre)改密

Dockerfile

@@ -37,6 +37,12 @@ COPY --from=stage-build /opt/jumpserver/release/jumpserver /opt/jumpserver
 RUN mkdir -p /root/.ssh/ \
     && echo "Host *\n\tStrictHostKeyChecking no\n\tUserKnownHostsFile /dev/null" > /root/.ssh/config
 
+RUN mkdir -p /opt/jumpserver/oracle/
+ADD https://f2c-north-rel.oss-cn-qingdao.aliyuncs.com/2.0/north/jumpserver/instantclient-basiclite-linux.x64-21.1.0.0.0.tar /opt/jumpserver/oracle/
+RUN tar -xvf /opt/jumpserver/oracle/instantclient-basiclite-linux.x64-21.1.0.0.0.tar
+RUN sh -c "echo /opt/jumpserver/oracle/instantclient_21_1 > /etc/ld.so.conf.d/oracle-instantclient.conf"
+RUN ldconfig
+
 RUN echo > config.yml
 VOLUME /opt/jumpserver/data
 VOLUME /opt/jumpserver/logs

apps/assets/migrations/0077_auto_20210802_1415.py

+# Generated by Django 3.1.6 on 2021-08-02 06:15
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('applications', '0009_applicationuser'),
+        ('assets', '0076_delete_assetuser'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='authbook',
+            name='database',
+            field=models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, to='applications.application', verbose_name='Database'),
+        ),
+        migrations.AddField(
+            model_name='historicalauthbook',
+            name='database',
+            field=models.ForeignKey(blank=True, db_constraint=False, null=True, on_delete=django.db.models.deletion.DO_NOTHING, related_name='+', to='applications.application', verbose_name='Database'),
+        ),
+        migrations.AlterField(
+            model_name='authbook',
+            name='asset',
+            field=models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, to='assets.asset', verbose_name='Asset'),
+        ),
+    ]

apps/assets/models/authbook.py

@@ -12,7 +12,8 @@ __all__ = ['AuthBook']
 
 
 class AuthBook(BaseUser, AbsConnectivity):
-    asset = models.ForeignKey('assets.Asset', on_delete=models.CASCADE, verbose_name=_('Asset'))
+    asset = models.ForeignKey('assets.Asset', on_delete=models.CASCADE, verbose_name=_('Asset'), null=True)
+    database = models.ForeignKey('applications.Application', on_delete=models.CASCADE, null=True, verbose_name=_('Database'))
     systemuser = models.ForeignKey('assets.SystemUser', on_delete=models.CASCADE, null=True, verbose_name=_("System user"))
     version = models.IntegerField(default=1, verbose_name=_('Version'))
     history = HistoricalRecords()

apps/assets/signals_handler/authbook.py

@@ -39,4 +39,7 @@ def on_authbook_pre_create(sender, instance, **kwargs):
     # 升级版本号
     instance.version = instance.history.all().count() + 1
     # 即使在 root 组织也不怕
-    instance.org_id = instance.asset.org_id
+    if instance.database:
+        instance.org_id = instance.database.org_id
+    else:
+        instance.org_id = instance.asset.org_id

requirements/requirements.txt

@@ -113,4 +113,7 @@ termcolor==1.1.0
 azure-identity==1.5.0
 azure-mgmt-subscription==1.0.0
 qingcloud-sdk==1.2.12
-django-simple-history==3.0.0
\ No newline at end of file
+django-simple-history==3.0.0
+PyMySQL==1.0.2
+cx-Oracle==8.2.1
+psycopg2-binary==2.9.1