# -*- 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()) if u'电信' in row[7] : 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 = {} for iccid in iccids: devObjs = Device.objects.filter(iccid__contains = iccid,ownerId__ne = None,simExpireDate=None) 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 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 print 'finished. %s' % count