def test_del_batch_manufacturer(self):
manufacturers = user_api.find_manufacturers_by_query_terms({'query_box': 'mmm'})
manufacturer_ids = [manufacturer.id for manufacturer in manufacturers]
user_api.del_batch_manufacturer(manufacturer_ids)
manufacturers = user_api.find_manufacturers_by_query_terms({'query_box': 'mmm'})
self.assertEqual(len(manufacturers), 0)
def test_del_manufacturer(self):
manufacturers = user_api.find_manufacturers_by_query_terms({'query_box': 'mmm0'})
manufacturer = manufacturers[0]
user_api.del_manufacturer(manufacturer.id)
manufacturers = user_api.find_manufacturers_by_query_terms({'query_box': 'mmm0'})
self.assertEqual(len(manufacturers), 0)
def test_edit_manufacturer(self):
manufacturers = user_api.find_manufacturers_by_query_terms({'query_box': 'mmm0'})
manufacturer = manufacturers[0]
manufacturer_dict = manufacturer.get_dict()
manufacturer_dict['name'] = 'new' + manufacturer.name
manufacturer_dict.pop('remark')
manufacturer_dict.pop('create_time')
user_api.edit_manufacturer(manufacturer_dict)
manufacturers = user_api.find_manufacturers_by_query_terms({'query_box': 'mmm0'})
manufacturer = manufacturers[0]
self.assertEqual(manufacturer.name, 'newmmm0')
manufacturer_dict['phone'] = '13850592084'
error = False
try:
user_api.edit_manufacturer(manufacturer_dict)
except NotUniqueException:
error = True
self.assertTrue(error)
error = False
manufacturer_dict['id'] = 1000
try:
user_api.edit_manufacturer(manufacturer_dict)
except DoesNotExistException:
error = True
self.assertTrue(error)
def test_find_manufacturers_by_query_terms(self):
manufacturers = user_api.find_manufacturers_by_query_terms({})
self.assertEqual(len(manufacturers), 10)
manufacturers = user_api.find_manufacturers_by_query_terms({'query_box': 'mmm'})
self.assertEqual(len(manufacturers), 10)
manufacturers = user_api.find_manufacturers_by_query_terms({'query_box': 'bbb'})
self.assertEqual(len(manufacturers), 0)
manufacturers = user_api.find_manufacturers_by_query_terms({'query_box': '23477'})
self.assertEqual(len(manufacturers), 1)
manufacturers = user_api.find_manufacturers_by_query_terms({'query_box': '138'})
self.assertEqual(len(manufacturers), 10)
manufacturers = user_api.find_manufacturers_by_query_terms(
{'end_time': int(time.mktime(datetime.datetime(2020, 2, 3).timetuple()))})
self.assertEqual(len(manufacturers), 3)
manufacturers = user_api.find_manufacturers_by_query_terms(
{'begin_time': int(time.mktime(datetime.datetime(2020, 2, 3).timetuple()))})
self.assertEqual(len(manufacturers), 8)
manufacturers = user_api.find_manufacturers_by_query_terms(
{'begin_time': int(time.mktime(datetime.datetime(2020, 2, 3).timetuple())),
'end_time': int(time.mktime(datetime.datetime(2020, 2, 5).timetuple()))})
self.assertEqual(len(manufacturers), 3)
def query_manufacturer(self, request):
param = request_api.get_param(request)
page = request_api.get_page(request)
manufacturers, num = app_user_api.find_manufacturers_by_query_terms(
param, page)
return Result.success(data=manufacturers, count=num)
def test_manufacturer_import_and_export(self):
manufacturers = user_api.find_manufacturers_by_query_terms({'query_box': 'aaa'})
# 开始导出
user_api.manufacturer_export({'query_box': 'mmm'}, '/manufacturer.xls')
manufacturer_ids = [manufacturer.id for manufacturer in manufacturers]
user_api.del_batch_manufacturer(manufacturer_ids)
# 测试是否被删除成功
manufacturers = user_api.find_manufacturers_by_query_terms({'query_box': 'mmm'})
self.assertEqual(len(manufacturers), 0)
# 开始导入
user_api.manufacturer_import('/manufacturer.xls')
# 测试是否导入成功
manufacturers = user_api.find_manufacturers_by_query_terms({'query_box': 'mmm'})
self.assertEqual(len(manufacturers), 10)
# 清理生成的额外文件
del_file('/manufacturer.xls')
def find_dtu_regions_by_query_terms(query_terms, page=None):
# 查找指定条件供气商
mans, num = app_user_api.find_manufacturers_by_query_terms(query_terms)
# 提取所有供气商的id
man_ids = [m['id'] for m in mans]
if len(man_ids) == 0:
return []
# 构造查询框的查询条件
query_box = QueryTerms.make_or_query_terms(manufacturer_id=man_ids)
regions, num = conf_region_api.find_regions(query_box.get_filters(), page)
return regions, num