manyi/script/simCard/find_dianxin_expired_card2.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())
        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