def test_sort_ascending(self): base_data = Data.get_month_data(2000, 1) data = Data.sort_ascending(base_data) expects = [ "給与", "コンビニ", "その他1", "必需品1", "必需品2", "現金収入", "銀行収入", "スーパー", "計算外", "貯金", "PayPayチャージ", "PayPayチャージ", "電気代", "ガス代", "水道代", "立替分1", "立替分2" ] self._assert_list(data, expects)
def test_sort_ascending_nothing(self): data = Data.get_month_data(1999, 1) sorted_data = Data.sort_ascending(data) self.assertEqual(sorted_data.count(), 0)
def get(self, request, *args, **kwargs): context = { 'app_name': settings.APP_NAME, 'username': request.user, 'directions': Direction.list(), 'methods': Method.list(), 'unused_methods': Method.un_used_list(), 'first_categories': Category.first_list(), 'latter_categories': Category.latter_list(), 'temps': { 0: "No", 1: "Yes" }, 'checkeds': { 0: "No", 1: "Yes" }, } # 入力値を維持するための変数 query_context = {} query_list = [ "start_year", "start_month", "start_day", "end_year", "end_month", "end_day", "item", "lower_price", "upper_price" ] for q in query_list: if q in request.GET: query_context[q] = request.GET.get(q) context.update(query_context) query_context = {} query_list = ["direction", "method", "category", "temp", "checked"] for q in query_list: if q in request.GET and request.GET.get(q) != '': query_context[q] = list(map(int, request.GET.getlist(q))) context.update(query_context) # 条件に沿うデータを検索 if "is_query" in request.GET: # 日付 start_date = None if "start_year" in request.GET and "start_month" in request.GET \ and "start_day" in request.GET: if request.GET.get("start_year") != "" \ and request.GET.get("start_month") != "" \ and request.GET.get("start_day") != "": try: date(int(request.GET.get("start_year")), int(request.GET.get("start_month")), int(request.GET.get("start_day"))) start_date = request.GET.get("start_year") + "-" + request.GET.get("start_month") + \ "-" + request.GET.get("start_day") except: start_date = None end_date = None if "end_year" in request.GET and "end_month" in request.GET and "end_day" in request.GET: if request.GET.get("end_year") != "" and request.GET.get( "end_month") != "" and request.GET.get( "end_day") != "": try: date(int(request.GET.get("end_year")), int(request.GET.get("end_month")), int(request.GET.get("end_day"))) end_date = request.GET.get( "end_year") + "-" + request.GET.get( "end_month") + "-" + request.GET.get("end_day") except: end_date = None data = Data.get_range_data(start_date, end_date) # 品目 if "item" in request.GET: item = request.GET.get("item") if item != "": data = Data.get_keyword_data(data, item) # 金額 if "lower_price" in request.GET and request.GET.get( "lower_price") != "": try: data = Data.filter_price( data, int(request.GET.get("lower_price")), None) except: data = data if "upper_price" in request.GET and request.GET.get( "upper_price") != "": try: data = Data.filter_price( data, None, int(request.GET.get("upper_price"))) except: data = data # 方向 if "direction" in request.GET: try: data = Data.filter_directions(data, context["direction"]) except: data = data # 支払い方法 if "method" in request.GET: try: data = Data.filter_methods(data, context["method"]) except: data = data # 分類 if "category" in request.GET: try: data = Data.filter_categories(data, context["category"]) except: data = data # 立替 if "temp" in request.GET: try: data = Data.filter_temps(data, context["temp"]) except: data = data # チェック済み if "checked" in request.GET: try: data = Data.filter_checkeds(data, context["checked"]) except: data = data # 日付順にソート data = Data.sort_ascending(data) context.update({ "show_data": data, "income_sum": Data.get_income_sum(data), "outgo_sum": Data.get_outgo_sum(data), "is_show": True, }) else: context.update({"is_show": False}) return render(request, 'search.html', context)