# -*- 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 = [] iccidDict = {} dxIccids = [] # 电信的iccid,不一样哟。设备报上来的比excel的多一位 for i in range(1, nrows): row = sheet.row_values(i) try: tempStr = row[1].replace('\t','').upper() iccidDict[tempStr] = to_datetime('2022-04-30 23:59:59') tempStr = row[1].replace('\t','').lower() iccidDict[tempStr] = to_datetime('2022-04-30 23:59:59') 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 = iccidDict.keys(),ownerId__ne = None) devNos = [] for devObj in devObjs: # print iccidDict[devObj.iccid],devObj.simExpireDate if not devObj.simExpireDate or not iccidDict[devObj.iccid]: continue deltaDay = (devObj.simExpireDate - iccidDict[devObj.iccid]).days if deltaDay>15: print devObj.devNo,iccidDict[devObj.iccid],devObj.simExpireDate 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 23:59:59') # 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