def fetch_new_alarms():
fetch_count = 0
alarms = Alarm.fetch_online(cancel=False)
if Alarm.objects.filter(url=alarms[0]['url']).count() > 0:
return 0
all_areas = City.get_all_areas()
cities_below = {}
for am in reversed(alarms):
#print am['color_id'],am['title'],am['url'],am['pub_time']
url = am['url']
if Alarm.objects.filter(url=url).count()>0:
continue
alarm_obj =Alarm()
for key,val in am.iteritems():
setattr(alarm_obj,key,val)
title = am['title']
for code,area in all_areas:
if area in title:
alarm_obj.area_code = code
break
else:
raise ValueError(u'%s failed to match any areas' % title)
#匹配城市
if len(code) < 9:
title = title[len(area):-1]
if not cities_below.has_key(code):
cities_below[code] = City.get_below_cities(code)
for cid,city in cities_below[code]:
#print code,cid,city
if city in title:
alarm_obj.area_code = cid
break
#print alarm_obj.area_code,alarm_obj.title
alarm_obj.save()
fetch_count+=1
Log(level=1,event="new alarm:%s-%s(%s)" % (
alarm_obj.title,alarm_obj.pub_time,alarm_obj.area_code)).save()
return fetch_count
def send_alarm_sms():
#组装飞信发送列表,测试数据为2012月7月2日12时
clock = datetime(2012,7,2,16,0,0,0) if MODE == 'TEST' else current_clock_dt()
current_alarms = Alarm.objects.filter(pub_time=clock)
to_send_list = []
for current_alarm in current_alarms :
# if is_ignore_city(current_alarm.title):
# continue
'''省级预警存在市级行政区字样时忽略'''
if len(current_alarm.area_code) == 5 and in_ignore_text(current_alarm.title):
continue
users = User.objects.filted_with_cid(current_alarm.area_code)
if (len(users)>0 and current_alarm.content == ''):
current_alarm.content = Alarm.fetch_content(current_alarm.url)
current_alarm.save()
for user in users:
try:
alarm_log = AlarmLog.objects.get(alarm=current_alarm,user=user)
continue
except:
to_send_list.append((user,current_alarm))
#开始发送（单线程）
send_count = 0
if len(to_send_list) > 0 :
for i in range(0,LOGIN_LIMIT):
try:
ft = MyFetion(PHONE,PSW)
break
except:i+=1
else:
Log(level=1,event="failed to initial MyFetion in sending sms")
sys.exit(1)
#这里可以一次最大发送量
for user,alarm in to_send_list[0:SEND_MAX]:
fid = MyFetion.test_id if MODE == 'TEST' else user.fid
tail = u'via飞信天气网(预警短信Beta)'
content = u'[%s]\n%s %s' % (alarm.title,alarm.content,tail)
#测试和开发时给自己发送
#if MODE != 'PRODUCT':pass
details = u'status:ready %s-%s:%s' %(user.phone_num,alarm.area_code,content)
#保证先写入库成功再发送，不会造成重复发送，入库时失败，不会向该用户发送
#但是存在一个问题，即入库成功，但发送失败，即下面的sendBYid失败后，不会再向该用户发送
#如果脚本在sendBYid中断,日志信息(details列)也不会获得更新，即使是发送成功，也会显示status:failed。
try:
AlarmLog(alarm=alarm,user=user,details=details).save()
except Exception,e :
error_msg = 'alarmlog write to db error:%s:%s:%s' % (e,alarm,user)
Log(level=0,event = error_msg).save()
if MODE != 'PRODUCT': print error_msg
else:
send_result = ft.sendBYid(fid, content.encode('utf-8'))
if send_result == True or send_result == None:
send_count +=1
did = "OK" if send_result == True else "Failed"
details = 'status:%s,%s,area_code:%s,content:%s' % (
did,user.phone_num,alarm.area_code,content)
AlarmLog.objects.filter(
alarm=alarm,user=user).update(details=details)
#将发送结果发给自己
ft.send2self(details.encode('utf-8'))
if MODE != 'PRODUCT': print user.phone_num,content
ft.logout()
