manyi/script/simCard/find_dianxin_expired_card.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
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