34 lines
1.3 KiB
Python
34 lines
1.3 KiB
Python
from .models import Equipment
|
|
from apps.system.models import Dept
|
|
from apps.mtm.models import Mgroup
|
|
|
|
def daoru_equipment(path:str):
|
|
from apps.utils.snowflake import idWorker
|
|
from openpyxl import load_workbook
|
|
wb = load_workbook(path)
|
|
sheet = wb.get_sheet_by_name('em_equipment')
|
|
i = 2
|
|
while sheet[f'a{i}'].value:
|
|
name = sheet[f'a{i}'].value
|
|
number = sheet[f'b{i}'].value
|
|
model = sheet[f'c{i}'].value
|
|
parameter = sheet[f'd{i}'].value
|
|
place = sheet[f'e{i}'].value
|
|
power_kw = sheet[f'f{i}'].value
|
|
belong_dept_name = sheet[f'g{i}'].value
|
|
mgroup_name = sheet[f'h{i}'].value
|
|
Equipment.objects.get_or_create(
|
|
number=number,
|
|
defaults={
|
|
'id': idWorker.get_id(),
|
|
'name': name,
|
|
'model': model,
|
|
'parameter': parameter,
|
|
'place': place,
|
|
'power_kw': power_kw,
|
|
'belong_dept': Dept.objects.get(name=belong_dept_name) if belong_dept_name else None,
|
|
'mgroup': Mgroup.objects.get(name=mgroup_name) if mgroup_name else None
|
|
}
|
|
)
|
|
i = i + 1
|