manyi/script/shard_test/test.py

# -*- coding: utf-8 -*-
#!/usr/bin/env python

import os, sys
import threading
import uuid
import xlrd
from xlrd import xldate_as_tuple
import simplejson as json
import datetime
from django.conf import settings

PROJECT_ROOT = os.path.join(os.path.abspath(os.path.split(os.path.realpath(__file__))[0] + "/.."), '..')
sys.path.insert(0, PROJECT_ROOT)

os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'configs.testing')

from script.base import init_env

init_env(interactive = False)


from apps.web.device.models import SIMCard,Device
from apps.web.constant import Const

from apps.web.user.models import ConsumeRecord

from pymongo import MongoClient
import time,datetime,random,copy
from bson import ObjectId

para = {
        "dealerId" : "5d9ec336003048b0f7fb399b",
        "packageId" : "2",
        "payOpenId" : "ojqSxwIlZTyVk0MzuBGBwf7tUwYw",
        "linkedRechargeRecordId" : ObjectId("5fcd000e6f29257970ba7513"),
        "orderNo" : "1607270419165316",
        "agentId" : "6417d4456f29257125ebf705"
    }

obj = {
    "orderNo" : "1607270419165316",
    "time" : "2020-12-07 00:00:19",
    "openId" : "ojqSxwIlZTyVk0MzuBGBwf7tUwYw",
    "nickname" : "云动月隐",
    "ownerId" : "5d9ec336003048b0f7fb399b",
    "coin" : 2,
    "money" : 2,
    "devNo" : "862285036672560",
    "devType" : "充电桩",
    "devTypeCode" : "",
    "logicalCode" : "400845",
    "groupId" : "5d9ec3ade305f75cd4d4a3e7",
    "address" : "育才路52号",
    "groupNumber" : "1",
    "groupName" : "52号",
    "isNormal" : True,
    "remarks" : "",
    "dateTimeAdded" : datetime.datetime.now(),
    "leftBalance" : "",
    "errorDesc" : "",
    "sequanceNo" : "",
    "startKey" : "ojqSxwIlZTyVk0MzuBGBwf7tUwYw-1607270416218-12801",
    "attachParas" : para,
    "servicedInfo" : {},
    "desc" : "",
    "notified" : True,
    "disputed" : False,
    "aggInfo" : {
        "duration" : 480
    },
    "rechargeRcdId" : "5fcd000e6f29257970ba7513",
    "payInfo" : [],
    "paymentInfo" : {},
    "package" : {},
    "association" : {}
}

def random_str(slen=28):
    seed = "1234567890abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!@#$%^&*()_+=-"
    sa = []
    for i in range(slen):
        sa.append(random.choice(seed))
    return ''.join(sa)

c = MongoClient(host="172.16.69.132", port=20000)
collection = c['washpay2']['ConsumeRecord']

curTime = time.time()

ownerList = ['5a153e808732d62e3647cfad','5a1795078732d671884755fc','5a1562f48732d65a3ddcd3e4']

while True:
    objList = []
    for ii in range(1000):
        ownerId = ownerList[random.randint(0,2)]#random_str(24)
        orderNo =  '%s%s' % (curTime,random_str(10))
        para.update({
                     'dealerId':ownerId,
                     'linkedRechargeRecordId':random_str(24),
                     'orderNo':orderNo,
                     'agentId':random_str(24)
                     })
        obj.update({
                    'orderNo':orderNo,
                    'openId':random_str(),
                    'ownerId':ownerId,
                    'devNo':random_str(15),
                    'logicalCode':'%s' % random.randint(0,65535),
                    'groupId':random_str(24),
                    'startKey':random_str(50),
                    'rechargeRcdId':random_str(24)
                    })
        objList.append(copy.deepcopy(obj))
    
    print('start insert now.....')
    startTime = time.time()
    for tmpObj in objList:
        collection.insert(tmpObj)
    
    print('spend time = %s' % (time.time() - startTime) )
    time.sleep(1)