程序扩展性的一个实例（下）

#! /usr/bin/env python
# -*- coding: utf-8 -*-
"""
@author: zhoujiebing
@contact: zhoujiebing@maimiaotech.com
@date: 2012-11-16 11:30
@version: 1.0.0
@license: Copyright maimiaotech.com
@copyright: Copyright maimiaotech.com

"""
import sys
sys.path.append('http://www.cnblogs.com/')
from shop_info_db import ShopInfoDB
from database_operator import DBOpsCampaignSelect, DBOpsItemsSelect, DBOpsKeyword, DBOpsActionLog, DBOpsShopInfo
MONGODB_HOST = 'localhost'
MONGODB_PORT = 2006
from pymongo import Connection
mongoConn = Connection(host = MONGODB_HOST, port = MONGODB_PORT)

class Rule:
"""
Base Rule Class
"""
def __init__(self):
self.child_rule = []

def judge_rule(self, shop_id, result):
return False

class AccessTokenRule(Rule):
"""
judge AccessToken
"""
def __init__(self, shop_status_dict):
Rule.__init__(self)
self.shop_status_dict = shop_status_dict

def judge_rule(self, shop_id, result):
shop_status = self.shop_status_dict[shop_id]
if shop_status['session_expired']:
resultinfo = ['Invalid AccessToken', shop_id]
result.append(resultinfo)
return True
if shop_status['insuff_level']:
resultinfo = ['R1 AccessToken', shop_id]
result.append(resultinfo)
return True
return False

class CampaignRule(Rule):
"""
judge campaign self
"""
def __init__(self):
Rule.__init__(self)

def judge_rule(self, shop_id, result):
campaign_select_db_ops = DBOpsCampaignSelect(mongoConn, str(shop_id))
campaigns = campaign_select_db_ops.get_all()
if len(campaigns) != 1:
resultinfo = ['No Campaign', shop_id]
result.append(resultinfo)
return True

if not campaigns[0]['init_success']:
resultinfo = ['Campaign init Fail', shop_id]
result.append(resultinfo)
return True
return False

class ItemRule(Rule):
"""
judge items
"""
def __init__(self):
Rule.__init__(self)

def judge_rule(self, shop_id, result):
item_select_db_ops = DBOpsItemsSelect(mongoConn, str(shop_id))
items_num = len(item_select_db_ops.get_active_items_all())
if items_num <= 10:
resultinfo = ['Item few', shop_id, str(items_num)]
result.append(resultinfo)
return True
return False

class KeywordRule(Rule):
"""
judge keyword
"""
def __init__(self):
Rule.__init__(self)

def judge_rule(self, shop_id, result):
keyword_db_ops = DBOpsKeyword(mongoConn, str(shop_id))
keywords_num = keyword_db_ops.coll.find().count()
if keywords_num <= 100:
resultinfo = ['Keyword few', shop_id, str(keywords_num)]
result.append(resultinfo)
return True
return False

class ActionRule(Rule):
"""
judge action
"""
def __init__(self):
Rule.__init__(self)

def judge_rule(self, shop_id, result):
action_db_ops = DBOpsActionLog(mongoConn, str(shop_id))
accumulate = action_db_ops.get_last_week_action()
action_num = accumulate['add_words'] + accumulate['delete_words'] + accumulate['change_price']
if action_num <= 100:
resultinfo = ['Action few', shop_id, str(action_num)]
return True
return False

def judge_rule(shop_id, unnormal_result, rule_relation):
"""
judge the shop
"""
while 1:
if rule_relation.judge_rule(shop_id, unnormal_result):
break
if not rule_relation.child_rule:
break
if len(rule_relation.child_rule) == 1:
rule_relation = rule_relation.child_rule[0]
else:
for rule_relation in rule_relation.child_rule:
if judge_rule(shop_id, unnormal_result, rule_relation):
return

if __name__ == '__main__':
file_name = 'unnormal_data.csv'
file_obj = file(file_name, 'w')

shop_info = ShopInfoDB(mongoConn)
shop_id_list = shop_info.get_all_shop_id_list()
shop_status_dict = {}
for shop in shop_info.shop_status.find():
shop_status_dict[int(shop['_id'])] = shop
#rule init
rule_relation = Rule()
access_token_rule = AccessTokenRule(shop_status_dict)
campaign_rule = CampaignRule()
item_rule = ItemRule()
keyword_rule = KeywordRule()
action_rule = ActionRule()

#rule relation init
rule_relation.child_rule.append(access_token_rule)
access_token_rule.child_rule.append(campaign_rule)
campaign_rule.child_rule.append(item_rule)
campaign_rule.child_rule.append(keyword_rule)
campaign_rule.child_rule.append(action_rule)

unnormal_result = []
for shop_id in shop_id_list:
judge_rule(shop_id, unnormal_result, rule_relation)
unnormal_result.sort(key = lambda x:x[0])
for info in unnormal_result:
shop_id = info[1]
info[1] = str(shop_id)
nick = shop_status_dict[shop_id]['nick']
info.insert(2, nick)
file_obj.write(','.join(info))
file_obj.write('\n')
file_obj.close()