MchInfoEncryptLogInfo

Brightcells 7 years ago
parent
commit
15e0f2277f

+ 30 - 12
api/encrypt_views.py

@@ -7,6 +7,7 @@ import random
7 7
 from django_logit import logit
8 8
 from django_response import response
9 9
 
10
+from logs.models import MchInfoEncryptLogInfo
10 11
 from mch.models import BrandInfo, ModelInfo
11 12
 from utils.algorithm.b64 import b64_decrypt, b64_encrypt
12 13
 from utils.algorithm.caesar import caesar_decrypt, caesar_encrypt
@@ -26,23 +27,40 @@ CIPHER_PREFIX = {
26 27
 @logit(res=True)
27 28
 def encrypt(request):
28 29
     plaintext = request.POST.get('plaintext', '')
30
+    optor_id = request.POST.get('optor_id', '')
29 31
 
30
-    alg = random.choice(CIPHER_ALGORITHM)
32
+    # brand_id#model_id#distributor_id#sn#time
33
+    # AAAA#AAAAAA#AAAAA#AAAAAAAAAAAAAA#180224
34
+    brand_pk, model_pk, distributor_pk, sn, time = plaintext.split('#')
31 35
 
32
-    if alg == 'CAESAR':
33
-        ciphertext = caesar_encrypt(plaintext)
34
-    elif alg == 'B64':
35
-        ciphertext = b64_encrypt(plaintext)
36
-    elif alg == 'RSA':
37
-        ciphertext = rsa_encrypt(plaintext)
38
-    else:
39
-        ciphertext = plaintext
36
+    mieli, created_at = MchInfoEncryptLogInfo.objects.get_or_create(plaintext=plaintext)
37
+
38
+    if created_at:
39
+        alg = random.choice(CIPHER_ALGORITHM)
40
+
41
+        if alg == 'CAESAR':
42
+            ciphertext = caesar_encrypt(plaintext)
43
+        elif alg == 'B64':
44
+            ciphertext = b64_encrypt(plaintext)
45
+        elif alg == 'RSA':
46
+            ciphertext = rsa_encrypt(plaintext)
47
+        else:
48
+            ciphertext = plaintext
49
+
50
+        mieli.alg = alg
51
+        mieli.ciphertext = ciphertext
52
+        mieli.brand_pk = brand_pk
53
+        mieli.model_pk = model_pk
54
+        mieli.distributor_pk = distributor_pk
55
+        mieli.sn = sn
56
+        mieli.operator_id = optor_id
57
+        mieli.save()
40 58
 
41 59
     return response(200, data={
42 60
         'ciphertext': u'{prefix}+{cipherlen}+{ciphertext}'.format(
43
-            prefix=CIPHER_PREFIX.get(alg, ''),
44
-            cipherlen=len(ciphertext),
45
-            ciphertext=ciphertext,
61
+            prefix=CIPHER_PREFIX.get(mieli.alg, ''),
62
+            cipherlen=len(mieli.ciphertext),
63
+            ciphertext=mieli.ciphertext,
46 64
         ),
47 65
     })
48 66
 

+ 7 - 1
logs/admin.py

@@ -3,7 +3,12 @@
3 3
 from django.contrib import admin
4 4
 from django_admin import ReadOnlyModelAdmin
5 5
 
6
-from logs.models import MchLogInfo
6
+from logs.models import MchInfoEncryptLogInfo, MchLogInfo
7
+
8
+
9
+class MchInfoEncryptLogInfoAdmin(ReadOnlyModelAdmin, admin.ModelAdmin):
10
+    list_display = ('plaintext', 'alg', 'ciphertext', 'brand_pk', 'model_pk', 'distributor_pk', 'sn', 'operator_id', 'status', 'created_at', 'updated_at')
11
+    list_filter = ('alg', 'brand_pk', 'model_pk', 'distributor_pk', 'operator_id', 'status')
7 12
 
8 13
 
9 14
 class MchLogInfoAdmin(ReadOnlyModelAdmin, admin.ModelAdmin):
@@ -13,3 +18,4 @@ class MchLogInfoAdmin(ReadOnlyModelAdmin, admin.ModelAdmin):
13 18
 
14 19
 
15 20
 admin.site.register(MchLogInfo, MchLogInfoAdmin)
21
+admin.site.register(MchInfoEncryptLogInfo, MchInfoEncryptLogInfoAdmin)

+ 35 - 0
logs/migrations/0003_mchinfoencryptloginfo.py

@@ -0,0 +1,35 @@
1
+# -*- coding: utf-8 -*-
2
+# Generated by Django 1.11.11 on 2018-06-07 10:21
3
+from __future__ import unicode_literals
4
+
5
+from django.db import migrations, models
6
+
7
+
8
+class Migration(migrations.Migration):
9
+
10
+    dependencies = [
11
+        ('logs', '0002_auto_20180530_1122'),
12
+    ]
13
+
14
+    operations = [
15
+        migrations.CreateModel(
16
+            name='MchInfoEncryptLogInfo',
17
+            fields=[
18
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
19
+                ('status', models.BooleanField(db_index=True, default=True, help_text='Status', verbose_name='status')),
20
+                ('created_at', models.DateTimeField(auto_now_add=True, help_text='Create Time', verbose_name='created_at')),
21
+                ('updated_at', models.DateTimeField(auto_now=True, help_text='Update Time', verbose_name='updated_at')),
22
+                ('plaintext', models.CharField(blank=True, db_index=True, help_text='\u5f85\u52a0\u5bc6\u5b57\u7b26\u4e32', max_length=64, null=True, unique=True, verbose_name='plaintext')),
23
+                ('alg', models.CharField(blank=True, help_text='\u52a0\u5bc6\u7b97\u6cd5', max_length=16, null=True, verbose_name='alg')),
24
+                ('ciphertext', models.CharField(blank=True, help_text='\u52a0\u5bc6\u5b57\u7b26\u4e32', max_length=64, null=True, verbose_name='ciphertext')),
25
+                ('brand_pk', models.IntegerField(db_index=True, default=0, help_text='\u54c1\u724cPK', verbose_name='brand_pk')),
26
+                ('model_pk', models.IntegerField(db_index=True, default=0, help_text='\u578b\u53f7PK', verbose_name='model_pk')),
27
+                ('sn', models.CharField(blank=True, db_index=True, help_text='\u5e8f\u5217\u53f7', max_length=32, null=True, verbose_name='sn')),
28
+                ('operator_id', models.CharField(blank=True, db_index=True, help_text='\u64cd\u4f5c\u5458\u552f\u4e00\u6807\u8bc6', max_length=32, null=True, verbose_name='operator_id')),
29
+            ],
30
+            options={
31
+                'verbose_name': 'mchinfoencryptloginfo',
32
+                'verbose_name_plural': 'mchinfoencryptloginfo',
33
+            },
34
+        ),
35
+    ]

+ 20 - 0
logs/migrations/0004_mchinfoencryptloginfo_distributor_pk.py

@@ -0,0 +1,20 @@
1
+# -*- coding: utf-8 -*-
2
+# Generated by Django 1.11.11 on 2018-06-07 10:29
3
+from __future__ import unicode_literals
4
+
5
+from django.db import migrations, models
6
+
7
+
8
+class Migration(migrations.Migration):
9
+
10
+    dependencies = [
11
+        ('logs', '0003_mchinfoencryptloginfo'),
12
+    ]
13
+
14
+    operations = [
15
+        migrations.AddField(
16
+            model_name='mchinfoencryptloginfo',
17
+            name='distributor_pk',
18
+            field=models.IntegerField(db_index=True, default=0, help_text='\u7ecf\u9500\u5546PK', verbose_name='distributor_pk'),
19
+        ),
20
+    ]

+ 22 - 0
logs/models.py

@@ -6,6 +6,28 @@ from django_models_ext import BaseModelMixin, upload_file_url, upload_path
6 6
 from shortuuidfield import ShortUUIDField
7 7
 
8 8
 
9
+class MchInfoEncryptLogInfo(BaseModelMixin):
10
+    plaintext = models.CharField(_(u'plaintext'), max_length=64, blank=True, null=True, help_text=u'待加密字符串', db_index=True, unique=True)
11
+
12
+    alg = models.CharField(_(u'alg'), max_length=16, blank=True, null=True, help_text=u'加密算法')
13
+    ciphertext = models.CharField(_(u'ciphertext'), max_length=64, blank=True, null=True, help_text=u'加密字符串')
14
+
15
+    brand_pk = models.IntegerField(_(u'brand_pk'), default=0, help_text=u'品牌PK', db_index=True)
16
+    model_pk = models.IntegerField(_(u'model_pk'), default=0, help_text=u'型号PK', db_index=True)
17
+    distributor_pk = models.IntegerField(_(u'distributor_pk'), default=0, help_text=u'经销商PK', db_index=True)
18
+
19
+    sn = models.CharField(_(u'sn'), max_length=32, blank=True, null=True, help_text=u'序列号', db_index=True)
20
+
21
+    operator_id = models.CharField(_(u'operator_id'), max_length=32, blank=True, null=True, help_text=u'操作员唯一标识', db_index=True)
22
+
23
+    class Meta:
24
+        verbose_name = _(u'mchinfoencryptloginfo')
25
+        verbose_name_plural = _(u'mchinfoencryptloginfo')
26
+
27
+    def __unicode__(self):
28
+        return unicode(self.pk)
29
+
30
+
9 31
 class MchLogInfo(BaseModelMixin):
10 32
     log_id = ShortUUIDField(_(u'log_id'), max_length=32, help_text=u'日志唯一标识', db_index=True)
11 33
     log_file = models.FileField(_(u'log_file'), upload_to=upload_path, blank=True, null=True, help_text=u'日志文件')

+ 1 - 1
requirements.txt

@@ -5,7 +5,7 @@ Pillow==5.0.0
5 5
 StatusCode==1.0.0
6 6
 TimeConvert==1.4.3
7 7
 cryptography==1.5.2
8
-django-admin==1.2.4
8
+django-admin==1.2.6
9 9
 django-cors-headers==2.2.0
10 10
 django-curtail-uuid==1.0.4
11 11
 django-detect==1.0.5