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