|
@@ -1,5 +1,7 @@
|
|
|
# coding=utf-8
|
|
|
import traceback
|
|
|
+import datetime
|
|
|
+import time
|
|
|
from django.db import transaction
|
|
|
from rest_framework.decorators import action
|
|
|
from rest_framework_jwt.views import ObtainJSONWebToken,VerifyJSONWebToken,RefreshJSONWebToken
|
|
@@ -77,8 +79,22 @@ class CompanyViewSet(CustomModelViewSet):
|
|
|
else:
|
|
|
raise CustomError('续费金额错误')
|
|
|
|
|
|
- result = []
|
|
|
try:
|
|
|
+ tenant = Tenant.objects.filter(id=pk).first()
|
|
|
+ if tenant:
|
|
|
+ today = datetime.datetime.now().strftime('%Y-%m-%d')
|
|
|
+ today_stamp = time.mktime(time.strptime(today,"%Y-%m-%d"))
|
|
|
+
|
|
|
+ end_date = time.mktime(time.strptime(tenant.end_date.strftime('%Y-%m-%d'), '%Y-%m-%d'))
|
|
|
+ diff = int(end_date) - int(today_stamp)
|
|
|
+ if diff >= 0:
|
|
|
+ start_date = tenant.end_date
|
|
|
+ renew_date = (tenant.end_date + datetime.timedelta(days=365*int(year))).strftime('%Y-%m-%d')
|
|
|
+ else:
|
|
|
+ start_date = today
|
|
|
+ renew_date = (datetime.datetime.now() + datetime.timedelta(days=365 * int(year))).strftime('%Y-%m-%d')
|
|
|
+ else:
|
|
|
+ raise CustomError('获取企业信息失败,请刷新重试!')
|
|
|
with transaction.atomic():
|
|
|
if int(pay_channel) == Pay.WECHAT:
|
|
|
line_data = Pay.wechatPay(request.user, pk ,amount)
|
|
@@ -87,6 +103,7 @@ class CompanyViewSet(CustomModelViewSet):
|
|
|
else:
|
|
|
raise CustomError('请选择正确的支付方式')
|
|
|
data = {
|
|
|
+ 'renew_date': '{} — {}'.format(start_date,renew_date),
|
|
|
'year': year,
|
|
|
'amount': Formater.formatAmountShow(amount),
|
|
|
'id': line_data.id,
|