专家分类表的建立与相关数据导入脚本

.gitignore

@@ -0,0 +1 @@
+/.vscode

Specialist/admin.py

@@ -0,0 +1,5 @@
+from django.contrib import admin
+from Specialist.models import SpecialistCategory
+
+# Register your models here.
+admin.site.register(SpecialistCategory)

Specialist/apps.py

@@ -0,0 +1,5 @@
+from django.apps import AppConfig
+
+
+class SpecialistConfig(AppConfig):
+    name = 'Specialist'

Specialist/migrations/0001_initial.py

@@ -0,0 +1,21 @@
+# Generated by Django 3.0 on 2019-12-16 03:10
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='SpecialistCategory',
+            fields=[
+                ('key', models.CharField(max_length=8, primary_key=True, serialize=False)),
+                ('name', models.CharField(max_length=20)),
+            ],
+        ),
+    ]

Specialist/models.py

@@ -0,0 +1,11 @@
+"""专家信息模型"""
+from django.db import models
+
+# Create your models here.
+class SpecialistCategory(models.Model):
+    """专家分类
+    - 分类key key char 8
+    - 分类名 name char 20
+    """
+    key = models.CharField(max_length=8, primary_key=True)
+    name = models.CharField(max_length=20)

Specialist/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

Specialist/views.py

@@ -0,0 +1,3 @@
+from django.shortcuts import render
+
+# Create your views here.

basic_scirpt/admin_signup.py

@@ -2,12 +2,11 @@
 """管理员注册脚本"""
 import sys
 import os
-import django
-
 import re
 import hashlib
 import getpass
 
+import django
 sys.path.append('../')
 os.environ['DJANGO_SETTINGS_MODULE'] = 'specialist_info.settings'
 django.setup()

basic_scirpt/specialist_type.py

@@ -0,0 +1,38 @@
+"""自动化生成数据库中的专家分类表"""
+#!/usr/bin/env python3
+import re
+import docx
+
+import sys
+import os
+import django
+sys.path.append('../')
+os.environ['DJANGO_SETTINGS_MODULE'] = 'specialist_info.settings'
+django.setup()
+
+from Specialist.models import SpecialistCategory
+
+def save(text):
+    """将传入的字符串格式化后存入数据库
+    分类ID: 如 A01 A0101
+    分类名称:
+    """
+    ret = re.match(r'^\w{1}\d{2,6}', text)
+    if ret:
+        key = ret.group()
+        val = re.split(r'^\w{1}\d{2,6} *', text)[1]
+        val = val.replace('\n', '').replace('\r', '')
+        obj = SpecialistCategory(key=key, name=val)
+        obj.save()
+
+def main():
+    """main"""
+    doc = docx.Document('type_info.docx')
+    for table in doc.tables:
+        for cell in table.columns[0].cells:
+            save(cell.text)
+        for cell in table.columns[1].cells:
+            save(cell.text)
+
+if __name__ == "__main__":
+    main()

specialist_info/settings.py

@@ -38,6 +38,7 @@
     'django.contrib.messages',
     'django.contrib.staticfiles',
     'UserInfo',
+    'Specialist',
 ]
 
 MIDDLEWARE = [