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)
