# -*- 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 from bson import ObjectId 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.production') from script.base import init_env init_env(interactive = False) from apps.web.device.models import SIMCard,Device from apps.web.dealer.models import Dealer from apps.web.constant import Const from apilib.utils_datetime import to_datetime from apps.web.user.models import RechargeRecord # 从SIM卡平台上导出来的excel,导入到数据库中,便于流量结算。 # 1、月末:根据用户的充值情况,把SIM卡数据全部导出来,发给卡商,进行充值。 # 2、月头:卡商充值后,我们利用import_sim_card脚本,把excel导出来,然后导入到数据库,并执行另外一个脚本update_device_sim_info_from_simdb更新设备的超时时间 supplier = sys.argv[1] excelFile = sys.argv[2] if supplier not in ['qiben','hezhou','jieyang','tianyu']: print u'卡供应商必须是qiben、hezhou' exit(0) book = xlrd.open_workbook(excelFile) try: sheet = book.sheet_by_name("sheet1") except Exception,e: print 'open excel file error =%s' % e exit(0) nrows = sheet.nrows rows = [] iccids = [] dxIccids = [] # 电信的iccid,不一样哟。设备报上来的比excel的多一位 for i in range(1, nrows): row = sheet.row_values(i) try: iccids.append(row[1].replace('\t','').upper()) iccids.append(row[1].replace('\t','').lower()) iccids.append(row[1].replace('\t','').upper() + '0') iccids.append(row[1].replace('\t','').lower() + '0') iccids.append(row[1].replace('\t','').upper() + '1') iccids.append(row[1].replace('\t','').lower() + '1') iccids.append(row[1].replace('\t','').upper() + '2') iccids.append(row[1].replace('\t','').lower() + '2') iccids.append(row[1].replace('\t','').upper() + '3') iccids.append(row[1].replace('\t','').lower() + '3') iccids.append(row[1].replace('\t','').upper() + '4') iccids.append(row[1].replace('\t','').lower() + '4') iccids.append(row[1].replace('\t','').upper() + '5') iccids.append(row[1].replace('\t','').lower() + '5') iccids.append(row[1].replace('\t','').upper() + '6') iccids.append(row[1].replace('\t','').lower() + '6') iccids.append(row[1].replace('\t','').upper() + '7') iccids.append(row[1].replace('\t','').lower() + '7') iccids.append(row[1].replace('\t','').upper() + '8') iccids.append(row[1].replace('\t','').lower() + '8') iccids.append(row[1].replace('\t','').upper() + '9') iccids.append(row[1].replace('\t','').lower() + '9') if row[7] == u'电信': dxIccids.append(row[1].replace('\t','').upper()) dxIccids.append(row[1].replace('\t','').lower()) except Exception,e: print 'some error when update iccid=%s,e=%s' % (row[1],e) devIccids = [] devList= [] count = 0 ownerDict = {} devObjs = Device.objects.filter(iccid__in = iccids,ownerId__ne = None,simExpireDate=None) devNos = [] for devObj in devObjs: # if str(devObj.ownerId) not in ownerDict: # ownerDict[str(devObj.ownerId)] = [devObj] # else: # ownerDict[str(devObj.ownerId)].append(devObj) # # count = count + 1 devNos.append(devObj.devNo) # for ownerId,devList in ownerDict.items(): # if not ownerId: # continue # owner = Dealer.objects.get(id = ObjectId(ownerId)) # for devObj in devList: # print owner.username,owner.nickname,ownerId,str(devObj.iccid).upper(),devObj.logicalCode,devObj.devNo count = 0 for devNo in devNos: count += 1 print devNo,count try: devObj = Device.objects(devNo = devNo).first() if devObj is None: continue devObj.simExpireDate = to_datetime('2022-04-30 00:00:00') devObj.simStatus = 'updated' devObj.simChargeAuto = True devObj.save() print 'updated one device OK,devNo=',devObj.devNo except Exception,e: continue Device.invalid_many_device_cache(devNos) # from apps.web.dealer.tasks import dealer_auto_charge_sim_card # dealer_auto_charge_sim_card() print 'finished. %s' % count