def search_for_actress(javlib_actress_code: str, page_num=1): search_url = 'star/{url_parameter}/{page_num}' db_conn = JavManagerDB() # get actress first page jav_objs, max_page = javbus_set_page(search_url, page_num=page_num, url_parameter=javlib_actress_code) for jav_obj in jav_objs: if db_conn.pk_exist(str(jav_obj.get('car'))): jav_obj.update(dict(db_conn.get_by_pk(str(jav_obj.get('car'))))) else: jav_obj['stat'] = 2 db_conn.upcreate_jav(jav_obj) return jav_objs, max_page
def get_set_javs(): set_type = request.args.get('set_type') page_num = request.args.get('page_num', 1) # optional search string search_string = request.args.get('search_string', '') _dedupe_car_list = [] rt_jav_objs = [] if not search_string: # parse set without search string # verify set type if set_type not in SET_TYPE_MAP: return jsonify( {'error': f'{set_type} is not a supported set type'}), 400 jav_objs, max_page = javbus_set_page(SET_TYPE_MAP[set_type], page_num) else: # search by supplied string search_map = { '番号': { 'function': search_by_car, 'params': { 'car': search_string } }, '女优': { 'function': search_for_actress, 'params': { 'javlib_actress_code': search_string, 'page_num': page_num } }, '分类': { 'function': javbus_set_page, 'params': { 'page_template': 'genre/{url_parameter}/{page_num}', 'page_num': page_num, 'url_parameter': search_string } }, } # verify set type if set_type not in search_map: return jsonify( {'error': f'{set_type} is not a supported search type'}), 400 jav_objs, max_page = search_map[set_type]['function']( **search_map[set_type]['params']) for jav_obj in jav_objs: if jav_obj['car'] not in _dedupe_car_list: _dedupe_car_list.append(jav_obj['car']) rt_jav_objs.append(jav_obj) # looooop through DB db_conn = JavManagerDB() for jav_obj in rt_jav_objs: if db_conn.pk_exist(str(jav_obj.get('car'))): jav_obj.update(dict(db_conn.get_by_pk(str(jav_obj.get('car'))))) else: jav_obj['stat'] = 2 db_conn.upcreate_jav(jav_obj) return jsonify( {'success': { 'jav_objs': rt_jav_objs, 'max_page': max_page }})