def set_is_valid():
args_spec = {
'district_id': Arg(int),
'district_ids': Arg(default=[]),
'is_valid': Arg(bool),
}
args = args_parser.parse(args_spec)
district_id = args['district_id']
district_ids = args['district_ids']
if district_id:
district_ids.append(district_id)
district_base.set_is_valid(district_ids, is_valid=args['is_valid'])
return ''
def set_is_valid():
args_spec = {
'district_id':Arg(int),
'district_ids':Arg(default=[]),
'is_valid':Arg(bool),
}
args = args_parser.parse(args_spec)
district_id = args['district_id']
district_ids = args['district_ids']
if district_id:
district_ids.append(district_id)
district_base.set_is_valid(district_ids,is_valid=args['is_valid'])
return ''
def post_or_put(pk=None, dic=None):
if pk is not None:
pk = int(pk)
# 处理新建与更新
new_city = dic
new_city['is_valid'] = True
check_same(new_city)
city_id = city_base.post_or_put(pk, new_city)
new_city['id'] = city_id
_districts = new_city.get('_districts', [])
for _district in _districts:
_district['is_valid'] = True
_district['city_id'] = new_city['id']
_district['id'] = district_base.post_or_put(_district['id'], _district)
_zones = _district.get('_zones', [])
for _zone in _zones:
_zone['is_valid'] = True
_zone['city_id'] = new_city['id']
_zone['district_id'] = _district['id']
_zone['id'] = zone_base.post_or_put(_zone['id'], _zone)
# 处理删除
if pk is not None:
old_city = get(pk)
old_districts = old_city.get('_districts', [])
old_district_ids = set([district['id'] for district in old_districts])
new_districts = new_city.get('_districts', [])
new_district_ids = set([district['id'] for district in new_districts])
delete_ids = list(old_district_ids - new_district_ids)
district_base.set_is_valid(delete_ids, False)
old_zones = reduce(
operator.add,
[_district.get('_zones', []) for _district in old_districts], [])
old_zone_ids = set([zone['id'] for zone in old_zones])
new_zones = reduce(
operator.add,
[_district.get('_zones', []) for _district in new_districts], [])
new_zone_ids = set([zone['id'] for zone in new_zones])
delete_ids = list(old_zone_ids - new_zone_ids)
zone_base.set_is_valid(delete_ids, False)
def post_or_put(pk=None, dic=None):
if pk is not None:
pk = int(pk)
# 处理新建与更新
new_city = dic
new_city['is_valid'] = True
check_same(new_city)
city_id = city_base.post_or_put(pk, new_city)
new_city['id'] = city_id
_districts = new_city.get('_districts', [])
for _district in _districts:
_district['is_valid'] = True
_district['city_id'] = new_city['id']
_district['id'] = district_base.post_or_put(_district['id'], _district)
_zones = _district.get('_zones', [])
for _zone in _zones:
_zone['is_valid'] = True
_zone['city_id'] = new_city['id']
_zone['district_id'] = _district['id']
_zone['id'] = zone_base.post_or_put(_zone['id'], _zone)
# 处理删除
if pk is not None:
old_city = get(pk)
old_districts = old_city.get('_districts', [])
old_district_ids = set([district['id'] for district in old_districts])
new_districts = new_city.get('_districts', [])
new_district_ids = set([district['id'] for district in new_districts])
delete_ids = list(old_district_ids - new_district_ids)
district_base.set_is_valid(delete_ids, False)
old_zones = reduce(
operator.add, [_district.get('_zones', []) for _district in old_districts], [])
old_zone_ids = set([zone['id'] for zone in old_zones])
new_zones = reduce(
operator.add, [_district.get('_zones', []) for _district in new_districts], [])
new_zone_ids = set([zone['id'] for zone in new_zones])
delete_ids = list(old_zone_ids - new_zone_ids)
zone_base.set_is_valid(delete_ids, False)
