manyi/script/shard_test/thread_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)

class Tester(threading.Thread):
    def __init__(self):
        super(Tester, self).__init__()
       
    def run(self):
        c = MongoClient(host="172.16.69.132", port=20000) #
        collection = c['washpay2']['ConsumeRecord']
        curTime = time.time()
        while True:
            objList = []
            for ii in range(1):
                ownerId = 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))
            
            startTime = time.time()
            for tmpObj in objList:
                collection.insert(tmpObj)
        
            print('spend time = %s' % (time.time() - startTime) )
            time.sleep(1)
                    
for ii in range(1):
    Tester().start()