youngS
4 years ago
parent
a82f2a3ec7
commit
250c07c1a6
@ -0,0 +1,47 @@ |
||||
# Generated by Django 3.0.3 on 2021-05-19 10:35 |
||||
|
||||
from django.conf import settings |
||||
from django.db import migrations, models |
||||
import django.db.models.deletion |
||||
|
||||
|
||||
class Migration(migrations.Migration): |
||||
dependencies = [ |
||||
('api', '0046_auto_20210423_1650'), |
||||
] |
||||
|
||||
operations = [ |
||||
migrations.CreateModel( |
||||
name='DomainCnameInfo', |
||||
fields=[ |
||||
('domain_record', |
||||
models.CharField(max_length=128, primary_key=True, serialize=False, verbose_name='记录值')), |
||||
('ip_address', models.GenericIPAddressField(verbose_name='域名解析地址')), |
||||
('is_enable', models.BooleanField(default=True, verbose_name='是否启用该解析')), |
||||
('description', models.TextField(blank=True, default='', null=True, verbose_name='备注')), |
||||
('created_time', models.DateTimeField(auto_now_add=True, verbose_name='创建时间')), |
||||
], |
||||
options={ |
||||
'verbose_name': '分发域名信息', |
||||
'verbose_name_plural': '分发域名信息', |
||||
}, |
||||
), |
||||
migrations.CreateModel( |
||||
name='UserDomainInfo', |
||||
fields=[ |
||||
('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')), |
||||
('domain_name', models.CharField(db_index=True, max_length=64, verbose_name='下载页面域名')), |
||||
('is_enable', models.BooleanField(default=False, verbose_name='绑定成功')), |
||||
('created_time', models.DateTimeField(auto_now_add=True, verbose_name='创建时间')), |
||||
('cname_id', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='api.DomainCnameInfo', |
||||
verbose_name='cname解析ID')), |
||||
('user_id', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL, |
||||
verbose_name='用户ID')), |
||||
], |
||||
options={ |
||||
'verbose_name': '分发域名绑定', |
||||
'verbose_name_plural': '分发域名绑定', |
||||
'unique_together': {('domain_name', 'is_enable')}, |
||||
}, |
||||
), |
||||
] |
||||
@ -0,0 +1,16 @@ |
||||
# Generated by Django 3.0.3 on 2021-05-19 15:53 |
||||
|
||||
from django.db import migrations |
||||
|
||||
|
||||
class Migration(migrations.Migration): |
||||
dependencies = [ |
||||
('api', '0047_domaincnameinfo_userdomaininfo'), |
||||
] |
||||
|
||||
operations = [ |
||||
migrations.AlterUniqueTogether( |
||||
name='userdomaininfo', |
||||
unique_together=set(), |
||||
), |
||||
] |
||||
@ -0,0 +1,16 @@ |
||||
# Generated by Django 3.0.3 on 2021-05-19 16:53 |
||||
|
||||
from django.db import migrations |
||||
|
||||
|
||||
class Migration(migrations.Migration): |
||||
dependencies = [ |
||||
('api', '0048_auto_20210519_1553'), |
||||
] |
||||
|
||||
operations = [ |
||||
migrations.RemoveField( |
||||
model_name='userinfo', |
||||
name='domain_name', |
||||
), |
||||
] |
||||
@ -0,0 +1,85 @@ |
||||
#!/usr/bin/env python |
||||
# -*- coding:utf-8 -*- |
||||
# project: 3月 |
||||
# author: NinEveN |
||||
# date: 2021/3/29 |
||||
|
||||
from rest_framework.views import APIView |
||||
|
||||
from api.utils.baseutils import is_valid_domain, get_cname_from_domain |
||||
from api.utils.response import BaseResponse |
||||
from api.utils.auth import ExpiringTokenAuthentication |
||||
from rest_framework.response import Response |
||||
from api.models import UserDomainInfo, DomainCnameInfo |
||||
from django.db.models import Count |
||||
import logging |
||||
|
||||
from api.utils.storage.caches import set_default_app_wx_easy |
||||
|
||||
logger = logging.getLogger(__name__) |
||||
|
||||
|
||||
class DomainCnameView(APIView): |
||||
authentication_classes = [ExpiringTokenAuthentication, ] |
||||
|
||||
def get(self, request): |
||||
res = BaseResponse() |
||||
res.data = {'domain_name': '', 'domain_record': '', 'is_enable': False} |
||||
user_domain_obj = UserDomainInfo.objects.filter(user_id=request.user).first() |
||||
if user_domain_obj: |
||||
res.data['domain_name'] = user_domain_obj.domain_name |
||||
res.data['is_enable'] = user_domain_obj.is_enable |
||||
if user_domain_obj.cname_id: |
||||
res.data['domain_record'] = user_domain_obj.cname_id.domain_record |
||||
return Response(res.dict) |
||||
|
||||
def post(self, request): |
||||
res = BaseResponse() |
||||
domain_name = request.data.get("domain_name", None) |
||||
if domain_name: |
||||
domain_name = domain_name.strip(" ") |
||||
if domain_name and len(domain_name) > 3 and is_valid_domain(domain_name): |
||||
if UserDomainInfo.objects.filter(domain_name=domain_name, is_enable=True).count() != 0: |
||||
res.code = 1001 |
||||
res.msg = "该域名已经被绑定，请更换其他域名" |
||||
else: |
||||
user_domian_obj = UserDomainInfo.objects.filter(user_id=request.user, domain_name=domain_name).first() |
||||
if user_domian_obj: |
||||
res.data = {'cname_domain': user_domian_obj.cname_id.domain_record} |
||||
else: |
||||
UserDomainInfo.objects.filter(user_id=request.user, is_enable=False).delete() |
||||
min_domian_cname_info_obj = min( |
||||
DomainCnameInfo.objects.annotate(Count('userdomaininfo')).filter(is_enable=True), |
||||
key=lambda x: x.userdomaininfo__count) |
||||
if min_domian_cname_info_obj: |
||||
res.data = {'cname_domain': min_domian_cname_info_obj.domain_record} |
||||
UserDomainInfo.objects.create(user_id=request.user, cname_id=min_domian_cname_info_obj, |
||||
domain_name=domain_name) |
||||
else: |
||||
res.code = 1002 |
||||
res.msg = "该域名校验失败，请检查" |
||||
return Response(res.dict) |
||||
|
||||
def put(self, request): |
||||
res = BaseResponse() |
||||
user_domian_obj = UserDomainInfo.objects.filter(user_id=request.user).first() |
||||
if user_domian_obj: |
||||
cname = get_cname_from_domain(user_domian_obj.domain_name) |
||||
if cname == user_domian_obj.cname_id.domain_record + '.': |
||||
user_domian_obj.is_enable = True |
||||
user_domian_obj.save() |
||||
UserDomainInfo.objects.filter(domain_name=user_domian_obj.domain_name, is_enable=False).delete() |
||||
set_default_app_wx_easy(request.user, True) |
||||
else: |
||||
res.code = 1003 |
||||
res.msg = "系统未检出到您的CNAME记录" |
||||
else: |
||||
res.code = 1004 |
||||
res.msg = "域名已经被绑定或者域名有误" |
||||
return Response(res.dict) |
||||
|
||||
def delete(sele, request): |
||||
res = BaseResponse() |
||||
if UserDomainInfo.objects.filter(user_id=request.user).delete()[0]: |
||||
set_default_app_wx_easy(request.user) |
||||
return Response(res.dict) |
||||
Loading…
Reference in new issue