def login_view(request):
username = request.data.get('username', '').lower()
password = request.data.get('password')
token = request.data.get('token')
try:
if token:
auth_user = Token.objects.get(key=token).user
else:
auth_user = authenticate(username=username, password=password)
if not auth_user and validate_email(username):
auth_user = get_user(email=username)
if not auth_user:
auth_user = authenticate(username=user.username, password=password)
if auth_user:
user = User.objects.get(auth_user=auth_user)
if user:
if auth_user.is_active:
response_data = UserSerializer(user).data
if Token.objects.filter(user=auth_user):
response_data['token'] = Token.objects.get(user=auth_user).key
return json_response(response_data, CODE_SUCCESS, MSG_LOGIN_SUCCESS)
else:
return invalid_token_response()
else:
return simple_json_response(CODE_NOT_ACTIVE, MSG_NOT_ACTIVE_USER)
else:
return simple_json_response(CODE_INCORRECT_USER_NAME_OR_PASSWORD, MSG_INCORRECT_USER_NAME_OR_PASSWORD)
else:
return simple_json_response(CODE_INCORRECT_USER_NAME_OR_PASSWORD, MSG_INCORRECT_USER_NAME_OR_PASSWORD)
except Exception as e:
return save_error_log(request, e)
def create(self, request, *args, **kwargs):
try:
category = request.data.get('category')
money = request.data.get('money')
remark = request.data.get('remark')
token = request.data.get('token')
date = request.data.get('date')
auth_user = get_user_by_token(token)
if auth_user:
iaer = Iaer()
iaer.user = User.objects.get(auth_user=auth_user)
iaer.money = money
iaer.category = category
iaer.remark = remark
iaer.created = timezone.now()
if int(money) > 0:
iaer.money_type = 1
else:
iaer.money_type = 0
try:
iaer.date = datetime.strptime(date, '%Y-%m-%d').date()
except ValueError:
print("date format error: %s" % date)
iaer.save()
response = IaerSerializer(iaer).data
return json_response(response, CODE_SUCCESS,
MSG_ADD_IAER_SUCCESS)
else:
return invalid_token_response()
except Exception as e:
return save_error_log(request, e)
def destroy(self, request, *args, **kwargs):
try:
token = request.data.get('token')
user = get_user_by_token(token)
if user:
red_envelope = self.get_object()
if red_envelope:
try:
response = super(RedEnvelopeViewSet,
self).destroy(request, *args,
**kwargs)
if response.status_code != status.HTTP_204_NO_CONTENT:
red_envelope.id = -1
except Exception as e:
red_envelope.id = -1
save_error_log(request, e)
event_json = RedEnvelopeSerializer(red_envelope).data
return json_response(event_json, CODE_SUCCESS,
MSG_DELETE_RED_ENVELOPE_SUCCESS)
else:
return simple_json_response(CODE_NO_CONTENT, MSG_204)
else:
return invalid_token_response()
except Exception as e:
return save_error_log(request, e)
def list(self, request, *args, **kwargs):
response_data = super(StatisticsCategoryViewSet,
self).list(request, *args, **kwargs).data
token = request.GET.get('token', '')
user = get_user_by_token(token)
response_data['year'] = int(request.GET.get(
'year', 0)) # if year = 0, get all category
response_data['month'] = int(request.GET.get(
'month',
0)) # if month = 0, get current year else get current month
for result in response_data['results']:
money = 0
if response_data['year'] == 0:
iaers = Iaer.objects.filter(Q(category = result['name']) & \
Q(user__id = user.id))
elif response_data['month'] == 0:
iaers = Iaer.objects.filter(Q(category = result['name']) & \
Q(user__id = user.id) & \
Q(created__year = response_data['year']))
else:
iaers = Iaer.objects.filter(Q(category = result['name']) & \
Q(user__id = user.id) & \
Q(created__year = response_data['year']) & \
Q(created__month = response_data['month']))
for iaer in iaers:
money += iaer.money
result['money'] = money
return json_response(response_data, CODE_SUCCESS,
MSG_GET_CATEGORIES_SUCCESS)
def update(self, request, *args, **kwargs):
token = request.data.get('token')
home_show_current = request.data.get('home_show_current', False)
home_show_this_month = request.data.get('home_show_this_month', False)
home_show_this_year = request.data.get('home_show_this_year', False)
monthly_fund = request.data.get('monthly_fund', 0)
yearly_fund = request.data.get('yearly_fund', 0)
try:
auth_user = get_user_by_token(token)
if auth_user:
user = User.objects.get(auth_user=auth_user)
if Setting.objects.filter(user_id=user.id):
setting = Setting.objects.get(user_id=user.id)
else:
setting = Setting()
setting.home_show_current = home_show_current
setting.home_show_this_month = home_show_this_month
setting.home_show_this_year = home_show_this_year
setting.monthly_fund = monthly_fund
setting.yearly_fund = yearly_fund
setting.user = user
setting.save()
return json_response(
SettingSerializer(setting).data, CODE_SUCCESS,
MSG_UPDATE_SETTING_SUCCESS)
else:
return invalid_token_response()
except Exception as e:
return save_error_log(request, e)
def destroy(self, request, *args, **kwargs):
try:
token = request.data.get('token')
auth_user = get_user_by_token(token)
if auth_user:
iaer = self.get_object()
if iaer:
try:
response = super(IaerViewSet,
self).destroy(request, *args,
**kwargs)
print("168: " + str(response.status_code))
if response.status_code != status.HTTP_204_NO_CONTENT:
iaer.id = -1
except Exception as e:
iaer.id = -1
save_error_log(request, e)
event_json = IaerSerializer(iaer).data
return json_response(event_json, CODE_SUCCESS,
MSG_DELETE_IAER_SUCCESS)
else:
return simple_json_response(CODE_NO_CONTENT, MSG_204)
else:
return invalid_token_response()
except Exception as e:
return save_error_log(request, e)
def retrieve(self, request, *args, **kwargs):
token = request.query_params.get('token')
try:
auth_user = get_user_by_token(token)
if auth_user:
response = super(UserViewSet, self).retrieve(request, *args, **kwargs).data
return json_response(response, CODE_SUCCESS, MSG_GET_USER_DETAIL_SUCCESS)
else:
return invalid_token_response()
except Exception as e:
return save_error_log(request, e)
def update(self, request, *args, **kwargs):
token = request.data.get('token')
user_name = request.data.get('user_name')
phone = request.data.get('phone')
email = request.data.get('email')
gender = request.data.get('gender')
birthday = request.data.get('birthday')
hobbies = request.data.get('hobbies')
base64 = request.data.get('base64')
try:
user = get_user_by_token(token)
if user:
if AuthUser.objects.filter(user=user):
user = AuthUser.objects.get(user=user)
if email:
if AuthUser.objects.filter(
username__iexact=email
) or AuthUser.objects.filter(email__iexact=email):
return simple_json_response(
CODE_DUPLICATE_USER, MSG_DUPLICATE_EMAIL)
else:
user.email = email
user.save()
if user_name:
user.user_name = user_name
if phone:
if AuthUser.objects.filter(phone=phone):
return simple_json_response(
CODE_DUPLICATE_PHONE, MSG_DUPLICATE_PHONE)
else:
user.phone = phone
if gender:
user.gender = gender
if birthday:
user.birthday = birthday
if hobbies:
user.hobbies = hobbies
if base64:
image_name = user.username + time.strftime(
'%Y%m%d%H%M%S') + PROFILE_FOOTER_IMAGE
user.save()
response_data = UserSerializer(user).data
response_data['token'] = Token.objects.get(user=user).key
return json_response(response_data, CODE_SUCCESS,
MSG_UPDATE_USER_INFO_SUCCESS)
return invalid_token_response()
except Exception as e:
return save_error_log(request, e)
def create(self, request, *args, **kwargs):
try:
serializer = self.get_serializer(data=request.data)
username = request.data.get('username')
user_name = request.data.get('user_name')
password = request.data.get('password')
email = request.data.get('email')
first_name = request.data.get('first_name', '')
last_name = request.data.get('last_name', '')
base64 = request.data.get('base64', '')
if not username:
return simple_json_response(CODE_EMPTY_USER, MSG_EMPTY_USERNAME)
elif not user_name:
return simple_json_response(CODE_EMPTY_BABY_NAME, MSG_EMPTY_BABY_NAME)
elif not email:
return simple_json_response(CODE_EMPTY_EMAIL, MSG_EMPTY_EMAIL)
elif not password:
return simple_json_response(CODE_EMPTY_PASSWORD, MSG_EMPTY_PASSWORD)
elif not validate_email(email):
return simple_json_response(CODE_INVALID_EMAIL, MSG_INVALID_EMAIL)
elif len(password) < MIN_PASSWORD_LEN:
return simple_json_response(CODE_INVALID_PASSWORD, MSG_INVALID_PASSWORD)
elif AuthUser.objects.filter(username__iexact=username) or AuthUser.objects.filter(username__iexact=email):
return simple_json_response(CODE_DUPLICATE_USER, MSG_DUPLICATE_USER)
elif AuthUser.objects.filter(email__iexact=email) or AuthUser.objects.filter(email__iexact=username):
return simple_json_response(CODE_DUPLICATE_EMAIL, MSG_DUPLICATE_EMAIL)
elif serializer.is_valid():
auth_user = AuthUser()
auth_user.email = email.lower()
auth_user.is_active = True
auth_user.is_staff = True
auth_user.set_password(password)
auth_user.username = username.lower()
auth_user.first_name = first_name
auth_user.last_name = last_name
self.request.user = auth_user
self.perform_create(serializer)
response_data = serializer.data
response_data['token'] = Token.objects.create(user=auth_user).key
user = User.objects.get(auth_user = auth_user)
user.locale = response_data['locale']
user.created = timezone.now()
if base64:
image_name = username + time.strftime('%Y%m%d%H%M%S') + PROFILE_FOOTER_IMAGE
user.save()
return json_response(response_data, CODE_SUCCESS, MSG_CREATE_USER_SUCCESS)
else:
return simple_json_response(CODE_INVALID_REQUEST, MSG_400)
except Exception as e:
return save_error_log(request, e)
def list(self, request, *args, **kwargs):
try:
serializer = self.get_serializer(data=request.data)
serializer.is_valid()
token = request.data.get('token')
auth_user = get_user_by_token(token)
if auth_user:
return json_response(super(UserViewSet, self).list(request, *args, **kwargs).data,
CODE_SUCCESS, MSG_GET_USERS_SUCCESS)
else:
return invalid_token_response()
except Exception as e:
return save_error_log(request, e)
def list(self, request, *args, **kwargs):
try:
token = request.query_params.get('token')
user = get_user_by_token(token)
if user:
year = datetime.now().year
month = datetime.now().month
response_data = super(IaerViewSet,
self).list(request, *args, **kwargs).data
income = 0
expenditure = 0
for iaer in self.get_queryset():
if iaer.money > 0:
income += iaer.money
else:
expenditure -= iaer.money
response_data['current_income'] = income
response_data['current_expenditure'] = expenditure
if int(self.request.query_params.get('top_list_size', 0)) == 0:
this_month_income = 0
this_month_expenditure = 0
for iaer in self.get_queryset().filter(
Q(date__year=year) & Q(date__month=month)):
if iaer.money > 0:
this_month_income += iaer.money
else:
this_month_expenditure -= iaer.money
response_data['this_month_income'] = this_month_income
response_data[
'this_month_expenditure'] = this_month_expenditure
this_year_income = 0
this_year_expenditure = 0
for iaer in self.get_queryset().filter(date__year=year):
if iaer.money > 0:
this_year_income += iaer.money
else:
this_year_expenditure -= iaer.money
response_data['this_year_income'] = this_year_income
response_data[
'this_year_expenditure'] = this_year_expenditure
return json_response(response_data, CODE_SUCCESS,
MSG_GET_IAERS_SUCCESS)
else:
return invalid_token_response()
except Exception as e:
return save_error_log(request, e)
def retrieve(self, request, *args, **kwargs):
if About.objects.all():
about = About.objects.all().order_by('-pk')[0]
else:
about = About()
about.id = -1
about.version_name = '1.0'
about.version_code = 1
about.category = -1
about.comment = ''
about.datetime = timezone.now()
return json_response(
AboutSerializer(about, context={
"request": request
}).data, CODE_SUCCESS, MSG_GET_ABOUT_SUCCESS)
def list(self, request, *args, **kwargs):
try:
token = request.query_params.get('token')
user = get_user_by_token(token)
if user:
total = 0
queryset = self.get_queryset()
for rer in queryset:
total += int(rer.money)
response_data = super(RedEnvelopeViewSet,
self).list(request, *args, **kwargs).data
response_data['total'] = total
return json_response(response_data, CODE_SUCCESS,
MSG_GET_RED_ENVELOPES_SUCCESS)
else:
return invalid_token_response()
except Exception as e:
return save_error_log(request, e)
def retrieve(self, request, *args, **kwargs):
token = request.query_params.get('token')
try:
auth_user = get_user_by_token(token)
if auth_user:
user = User.objects.get(auth_user=auth_user)
if Setting.objects.filter(user_id=user.id):
setting = Setting.objects.get(user_id=user.id)
else:
setting = Setting()
setting.created = timezone.now()
setting.user = user
setting.save()
return json_response(
SettingSerializer(setting).data, CODE_SUCCESS,
MSG_GET_SETTING_SUCCESS)
else:
return invalid_token_response()
except Exception as e:
return save_error_log(request, e)
def create(self, request, *args, **kwargs):
try:
money_from = request.data.get('money_from')
money = request.data.get('money')
remark = request.data.get('remark')
token = request.data.get('token')
user = get_user_by_token(token)
if user:
red_envelope = RedEnvelope()
red_envelope.user = User.objects.get(auth_user=user)
red_envelope.money = money
red_envelope.money_from = money_from
red_envelope.remark = remark
red_envelope.created = timezone.now()
red_envelope.save()
response = RedEnvelopeSerializer(red_envelope).data
return json_response(response, CODE_SUCCESS,
MSG_ADD_RED_ENVELOPE_SUCCESS)
else:
return invalid_token_response()
except Exception as e:
return save_error_log(request, e)
def list(self, request, *args, **kwargs):
return json_response(
super(FundViewSet, self).list(request, *args, **kwargs).data,
CODE_SUCCESS, MSG_GET_FUND_SUCCESS)
def list(self, request, *args, **kwargs):
response_data = super(StatisticsDateViewSet,
self).list(request, *args, **kwargs).data
token = request.GET.get('token', '')
user = get_user_by_token(token)
request_type = int(self.request.GET.get(
'type', 1)) # 1 for monthly, 2 for yearly
data_list = []
year = datetime.now().year
month = datetime.now().month
if user:
iaers = Iaer.objects.filter(user__id=user.id).order_by('created')
else:
iaers = Iaer.objects.filter(user__id=-1)
if request_type == 1:
for y in range(2017, year + 1): # get data from year 2017
for m in range(1, 12 + 1):
iaer_list = iaers.filter(
Q(created__year=y) & Q(created__month=m))
if len(iaer_list) > 0:
in_money = 0
out_money = 0
for iaer in iaer_list:
if iaer.money > 0:
in_money += iaer.money
else:
out_money += iaer.money
if in_money != 0:
data = OrderedDict([('year', y), ('month', m),
('money', in_money)])
data_list.append(data)
if out_money != 0:
data = OrderedDict([('year', y), ('month', m),
('money', out_money)])
data_list.append(data)
else:
for y in range(2017, year + 1): # get data from year 2017
iaer_list = iaers.filter(created__year=y)
if len(iaer_list) > 0:
in_money = 0
out_money = 0
for iaer in iaer_list:
if iaer.money > 0:
in_money += iaer.money
else:
out_money += iaer.money
if in_money != 0:
data = OrderedDict([('year', y), ('month', 0),
('money', in_money)])
data_list.append(data)
if out_money != 0:
data = OrderedDict([('year', y), ('month', 0),
('money', out_money)])
data_list.append(data)
response_data['results'] = data_list
response_data['count'] = len(data_list)
return json_response(response_data, CODE_SUCCESS,
MSG_GET_CATEGORIES_SUCCESS)
def list(self, request, *args, **kwargs):
return json_response(
super(CategoryViewSet, self).list(request, *args, **kwargs).data,
CODE_SUCCESS, MSG_GET_CATEGORIES_SUCCESS)
