You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
154 lines
5.1 KiB
154 lines
5.1 KiB
#!/usr/bin/env python
|
|
# -*- coding:utf-8 -*-
|
|
# project: 4月
|
|
# author: NinEveN
|
|
# date: 2021/4/16
|
|
|
|
import os, re, time
|
|
|
|
from django.db.models import Count
|
|
|
|
from fir_ser.settings import SUPER_SIGN_ROOT
|
|
from api.models import AppReleaseInfo, UserDomainInfo, DomainCnameInfo
|
|
from api.utils.app.randomstrings import make_app_uuid
|
|
from django.core.validators import validate_email
|
|
from django.core.exceptions import ValidationError
|
|
import logging
|
|
from dns.resolver import Resolver
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def get_app_d_count_by_app_id(app_id):
|
|
d_count = 1
|
|
binary_size = AppReleaseInfo.objects.filter(is_master=True, app_id__app_id=app_id).values('binary_size').first()
|
|
if binary_size and binary_size.get('binary_size', 0) > 0:
|
|
d_count += binary_size.get('binary_size') // 1024 // 1024 // 100
|
|
return d_count
|
|
|
|
|
|
def file_format_path(user_obj, auth=None, email=None):
|
|
if email:
|
|
cert_dir_name = make_app_uuid(user_obj, email)
|
|
else:
|
|
pkey = auth.get("username")
|
|
if auth.get("issuer_id"):
|
|
pkey = auth.get("issuer_id")
|
|
cert_dir_name = make_app_uuid(user_obj, pkey)
|
|
cert_dir_path = os.path.join(SUPER_SIGN_ROOT, cert_dir_name)
|
|
if not os.path.isdir(cert_dir_path):
|
|
os.makedirs(cert_dir_path)
|
|
file_format_path_name = os.path.join(cert_dir_path, cert_dir_name)
|
|
return file_format_path_name
|
|
|
|
|
|
def get_profile_full_path(developer_obj, app_obj):
|
|
pkey = developer_obj.email
|
|
if developer_obj.issuer_id:
|
|
pkey = developer_obj.issuer_id
|
|
cert_dir_name = make_app_uuid(developer_obj.user_id, pkey)
|
|
cert_dir_path = os.path.join(SUPER_SIGN_ROOT, cert_dir_name, "profile")
|
|
provision_name = os.path.join(cert_dir_path, app_obj.app_id)
|
|
return provision_name + '.mobileprovision'
|
|
|
|
|
|
def delete_app_profile_file(developer_obj, app_obj):
|
|
file = get_profile_full_path(developer_obj, app_obj)
|
|
try:
|
|
if os.path.isfile(file):
|
|
os.remove(file)
|
|
except Exception as e:
|
|
logger.error("delete_app_profile_file developer_obj:%s app_obj:%s file:%s Exception:%s" % (
|
|
developer_obj, app_obj, file, e))
|
|
|
|
|
|
def is_valid_domain(value):
|
|
pattern = re.compile(
|
|
r'^(([a-zA-Z]{1})|([a-zA-Z]{1}[a-zA-Z]{1})|'
|
|
r'([a-zA-Z]{1}[0-9]{1})|([0-9]{1}[a-zA-Z]{1})|'
|
|
r'([a-zA-Z0-9][-_.a-zA-Z0-9]{0,61}[a-zA-Z0-9]))\.'
|
|
r'([a-zA-Z]{2,13}|[a-zA-Z0-9-]{2,30}.[a-zA-Z]{2,3})$'
|
|
)
|
|
return True if pattern.match(value) else False
|
|
|
|
|
|
def is_valid_phone(value):
|
|
phone_pat = re.compile('^(13\d|14[5|7]|15\d|166|17[3|6|7]|18\d)\d{8}$')
|
|
return True if str(value) and re.search(phone_pat, str(value)) else False
|
|
|
|
|
|
def is_valid_email(email):
|
|
try:
|
|
validate_email(email)
|
|
return True
|
|
except ValidationError:
|
|
return False
|
|
|
|
|
|
def get_dict_from_filter_fields(filter_fields, data):
|
|
filter_data = {}
|
|
for filed in filter_fields:
|
|
f_value = data.get(filed, None)
|
|
if f_value:
|
|
filter_data[filed] = f_value
|
|
return filter_data
|
|
|
|
|
|
def format_storage_selection(storage_info_list, storage_choice_list):
|
|
storage_info_list.append({'id': -1, 'storage_type': 3, 'name': '默认存储'})
|
|
for storage_choice in storage_choice_list:
|
|
storage_choice['storage_info'] = []
|
|
for storage_info in storage_info_list:
|
|
if storage_info.get('storage_type') == storage_choice.get('id', ):
|
|
storage_choice['storage_info'].append(
|
|
{'name': storage_info.get('name', ''), 'id': storage_info.get('id', '')})
|
|
for storage_choice in storage_choice_list:
|
|
if not storage_choice['storage_info']:
|
|
storage_choice_list.remove(storage_choice)
|
|
return storage_choice_list
|
|
|
|
|
|
def get_cname_from_domain(domain):
|
|
dns_list = [
|
|
["8.8.8.8", "8.8.4.4"],
|
|
["119.29.29.29", "114.114.114.114"],
|
|
["223.5.5.5", "223.6.6.6"],
|
|
]
|
|
dns_resolver = Resolver()
|
|
domain = domain.lower().strip()
|
|
count = 3
|
|
while count:
|
|
try:
|
|
dns_resolver.nameservers = dns_list[len(dns_list) - count]
|
|
return dns_resolver.query(domain, 'CNAME')[0].to_text()
|
|
except Exception as e:
|
|
logger.error("dns %s resolve %s failed Exception:%s" % (dns_resolver.nameservers, domain, e))
|
|
count -= 1
|
|
time.sleep(0.3)
|
|
if count <= 0:
|
|
return str(None)
|
|
|
|
|
|
def get_user_domain_name(obj):
|
|
domain_obj = UserDomainInfo.objects.filter(user_id=obj, is_enable=True, app_id=None).first()
|
|
if domain_obj:
|
|
return domain_obj.domain_name
|
|
return ''
|
|
|
|
|
|
def get_app_domain_name(obj):
|
|
domain_obj = UserDomainInfo.objects.filter(app_id=obj, is_enable=True).first()
|
|
if domain_obj:
|
|
return domain_obj.domain_name
|
|
return ''
|
|
|
|
|
|
def get_user_default_domain_name(domain_cname_obj):
|
|
if domain_cname_obj:
|
|
return domain_cname_obj.is_https, domain_cname_obj.domain_record
|
|
return None, None
|
|
|
|
|
|
def get_min_default_domain_cname_obj(is_system=True):
|
|
return min(DomainCnameInfo.objects.annotate(Count('userdomaininfo')).filter(is_enable=True, is_system=is_system),
|
|
key=lambda x: x.userdomaininfo__count)
|
|
|