jupyter/体测单位/.ipynb_checkpoints/体测数据管理-checkpoint.ipynb

生成手机验证码

In [12]:

import openpyxl
import os,sys,shutil
import json
import math
import glob
import random
from pathlib import Path
import pymongo

myclient = pymongo.MongoClient('mongodb://localhost:27017/')
mydb = myclient["baogao"]

wb = openpyxl.load_workbook('data/北京石油分公司报告明细.xlsx')
place_id = 419309
dw_name = '北京石油分公司'
fi_path = 'file/2023-06-15'
fls = glob.glob(f'{fi_path}/*.pdf')

code_list = []
for i in range(10):   # 0~9
    code_list.append(str(i))
    
list1 = []
i = 0
while i < len(fls):
    code = random.sample(code_list,6)   #随机取6位数
    code_num = ''.join(code)
    if code_num not in list1:
        list1.append(code_num)
        i +=1
print(len(list1))
sheet = wb.active
# sheets = wb.sheetnames
list2 = []
i = 0
for n in range(2, sheet.max_row+1):
    if sheet.cell(n,1).value is not None:
        dict1 = {}
        phone = str(sheet.cell(n, 3).value)
        code = int(sheet.cell(n, 1).value)
        yzm = list1[i]
        i+=1
        dict1 = {'place_id':place_id,'code':yzm,'fn':str(code)+'.pdf'}
        list2.append(dict1)
mycol = mydb["pdf"]
x = mycol.insert_many(list2)

54

In [7]:

import openpyxl
import os,sys,shutil
import json
import math
import glob
import random
from pathlib import Path

wb = openpyxl.load_workbook('data/北京石油分公司报告明细.xlsx')
place_id = 1
dw_name = '北京石油分公司'
fi_path = 'file/2023-06-15'
fls = glob.glob(f'{fi_path}/*.pdf')

code_list = []
for i in range(10):   # 0~9
    code_list.append(str(i))
    
list1 = []
i = 0
while i < len(fls):
    code = random.sample(code_list,6)   #随机取6位数
    code_num = ''.join(code)
    if code_num not in list1:
        list1.append(code_num)
        i +=1
print(len(list1))

54

In [ ]: