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.
262 lines
10 KiB
262 lines
10 KiB
import logging
|
|
import os
|
|
|
|
from rest_framework import serializers
|
|
|
|
from api import models
|
|
from api.models import Apps
|
|
from api.utils.serializer import AppReleaseSerializer, UserInfoSerializer, AppsSerializer, StorageSerializer, \
|
|
ThirdWxSerializer, DomainNameSerializer, AppReportSerializer, OrdersSerializer
|
|
from common.base.baseutils import get_choices_dict, get_choices_name_from_key
|
|
from common.utils.caches import get_app_today_download_times
|
|
from common.utils.storage import Storage
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def get_download_url_from_context(self, obj, key, url, force_new=False):
|
|
download_url = ""
|
|
if self.context.get("key", None) and self.context.get("key") != "undefined":
|
|
key = self.context.get("key", '')
|
|
if self.context.get("storage", None) and self.context.get("storage") != "undefined":
|
|
storage = self.context.get("storage", None)
|
|
else:
|
|
if isinstance(obj, models.Apps):
|
|
storage = Storage(obj.user_id)
|
|
elif isinstance(obj, models.AppReleaseInfo):
|
|
storage = Storage(obj.app_id.user_id)
|
|
elif isinstance(obj, models.UserAdDisplayInfo):
|
|
storage = Storage(obj.user_id)
|
|
elif isinstance(obj, models.UserInfo):
|
|
storage = Storage(obj)
|
|
elif isinstance(obj, models.UserCertificationInfo) or isinstance(obj, models.CertificationInfo):
|
|
storage = Storage(obj.user_id, None, True)
|
|
else:
|
|
storage = None
|
|
if storage:
|
|
download_url = storage.get_download_url(os.path.basename(url), 600, key, force_new)
|
|
logger.info(f'get {os.path.basename(url)} download_url {download_url} force_new:{force_new} key:{key}')
|
|
return download_url
|
|
|
|
|
|
def get_app_master_obj_from_context(self, obj):
|
|
master_release_obj = models.AppReleaseInfo.objects.filter(app_id=obj, is_master=True).first()
|
|
if self.context.get("release_id", None) and self.context.get("release_id") != "undefined":
|
|
master_release_obj = models.AppReleaseInfo.objects.filter(app_id=obj,
|
|
release_id=self.context.get("release_id")).first()
|
|
return master_release_obj
|
|
|
|
|
|
def get_screenshots_from_self(self, obj, force_new=False):
|
|
screenshots_list = []
|
|
for screenshot_obj in models.AppScreenShot.objects.filter(app_id=obj).all():
|
|
icon_url = get_download_url_from_context(self, obj, '', screenshot_obj.screenshot_url, force_new)
|
|
screenshots_list.append({'id': screenshot_obj.pk, 'url': icon_url})
|
|
return screenshots_list
|
|
|
|
|
|
class AdminUserInfoSerializer(UserInfoSerializer):
|
|
class Meta:
|
|
model = models.UserInfo
|
|
exclude = ["password", "api_token"]
|
|
read_only_fields = ["id", "head_img", "free_download_times", "last_login",
|
|
"is_superuser", "last_name", "is_staff", "uid",
|
|
"date_joined", "download_times", "all_download_times", "storage", "groups",
|
|
"user_permissions", "certification_id", " app_count"]
|
|
|
|
gender_choices = serializers.SerializerMethodField()
|
|
|
|
def get_gender_choices(self, obj):
|
|
return get_choices_dict(obj.gender_choices)
|
|
|
|
role_choices = serializers.SerializerMethodField()
|
|
|
|
def get_role_choices(self, obj):
|
|
return get_choices_dict(obj.role_choices)
|
|
|
|
storage_choices = serializers.SerializerMethodField()
|
|
|
|
def get_storage_choices(self, obj):
|
|
return get_choices_dict(models.AppStorage.storage_choices[1:])
|
|
|
|
certification_status_choices = serializers.SerializerMethodField()
|
|
|
|
def get_certification_status_choices(self, obj):
|
|
return get_choices_dict(models.UserCertificationInfo.status_choices)
|
|
|
|
certification_id = serializers.SerializerMethodField()
|
|
|
|
def get_certification_id(self, obj):
|
|
return models.UserCertificationInfo.objects.filter(user_id=obj).values('id').first()
|
|
|
|
app_count = serializers.SerializerMethodField()
|
|
|
|
def get_app_count(self, obj):
|
|
return models.Apps.objects.filter(user_id=obj).count()
|
|
|
|
today_hits_count = serializers.SerializerMethodField()
|
|
|
|
def get_today_hits_count(self, obj):
|
|
android_app_ids = Apps.objects.filter(**{"user_id": obj, "type": 0}).values('app_id')
|
|
android_app_ids = [app_dict.get('app_id') for app_dict in android_app_ids]
|
|
ios_app_ids = Apps.objects.filter(**{"user_id": obj, "type": 1}).values('app_id')
|
|
ios_app_ids = [app_dict.get('app_id') for app_dict in ios_app_ids]
|
|
return {
|
|
"android_today_hits_count": get_app_today_download_times(android_app_ids),
|
|
"ios_today_hits_count": get_app_today_download_times(ios_app_ids)
|
|
}
|
|
|
|
def update(self, instance, validated_data):
|
|
return super(AdminUserInfoSerializer, self).update(instance, validated_data)
|
|
|
|
|
|
class AdminAppsSerializer(AppsSerializer):
|
|
class Meta:
|
|
model = models.Apps
|
|
fields = "__all__"
|
|
read_only_fields = ["id", "app_id", "user_id", "bundle_id", "count_hits", "updated_time", "created_time"]
|
|
|
|
status_choices = serializers.SerializerMethodField()
|
|
|
|
def get_status_choices(self, obj):
|
|
return get_choices_dict(obj.status_choices)
|
|
|
|
type_choices = serializers.SerializerMethodField()
|
|
|
|
def get_type_choices(self, obj):
|
|
return get_choices_dict(obj.type_choices)
|
|
|
|
supersign_type_choices = serializers.SerializerMethodField()
|
|
|
|
def get_supersign_type_choices(self, obj):
|
|
return get_choices_dict(obj.supersign_type_choices)
|
|
|
|
release_count = serializers.SerializerMethodField()
|
|
|
|
def get_release_count(self, obj):
|
|
return models.AppReleaseInfo.objects.filter(app_id=obj).count()
|
|
|
|
def update(self, instance, validated_data):
|
|
return super(AdminAppsSerializer, self).update(instance, validated_data)
|
|
|
|
|
|
class AdminAppReleaseSerializer(AppReleaseSerializer):
|
|
class Meta:
|
|
model = models.AppReleaseInfo
|
|
fields = '__all__'
|
|
read_only_fields = ["id", "app_id", "release_id", "binary_size", "udid", "icon_url"]
|
|
|
|
release_choices = serializers.SerializerMethodField()
|
|
app_aid = serializers.CharField(source='app_id.app_id', read_only=True)
|
|
|
|
def get_release_choices(self, obj):
|
|
return get_choices_dict(obj.release_choices)
|
|
|
|
def update(self, instance, validated_data):
|
|
print(validated_data)
|
|
if validated_data.get("is_master", False):
|
|
models.AppReleaseInfo.objects.filter(app_id=instance.app_id).update(**{"is_master": False})
|
|
else:
|
|
if "is_master" in validated_data and validated_data.get("is_master") != True:
|
|
del validated_data["is_master"]
|
|
return super(AdminAppReleaseSerializer, self).update(instance, validated_data)
|
|
|
|
|
|
class AdminStorageSerializer(StorageSerializer):
|
|
class Meta:
|
|
model = models.AppStorage
|
|
fields = "__all__"
|
|
read_only_fields = ["id", "user_id", "updated_time", "created_time", "storage_type"]
|
|
|
|
storage_choices = serializers.SerializerMethodField()
|
|
|
|
def get_storage_choices(self, obj):
|
|
return get_choices_dict(obj.storage_choices[1:])
|
|
|
|
used_id = serializers.IntegerField(source="user_id.pk")
|
|
|
|
|
|
class AdminOrdersSerializer(OrdersSerializer):
|
|
class Meta:
|
|
model = models.Order
|
|
fields = "__all__"
|
|
read_only_fields = ["id", "user_id", "updated_time", "created_time", "payment_number", "order_number",
|
|
"payment_name", "actual_amount", "actual_download_times", "actual_download_gift_times"]
|
|
|
|
payment_type_choices = serializers.SerializerMethodField()
|
|
|
|
def get_payment_type_choices(self, obj):
|
|
return get_choices_dict(obj.payment_type_choices)
|
|
|
|
status_choices = serializers.SerializerMethodField()
|
|
|
|
def get_status_choices(self, obj):
|
|
return get_choices_dict(obj.status_choices)
|
|
|
|
order_type_choices = serializers.SerializerMethodField()
|
|
|
|
def get_order_type_choices(self, obj):
|
|
return get_choices_dict(obj.order_type_choices)
|
|
|
|
|
|
class AdminUserCertificationSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = models.UserCertificationInfo
|
|
fields = "__all__"
|
|
read_only_fields = ["id", "user_id", "reviewed_time", "created_time"]
|
|
|
|
certification_status_choices = serializers.SerializerMethodField()
|
|
|
|
def get_certification_status_choices(self, obj):
|
|
return get_choices_dict(obj.status_choices)
|
|
|
|
certification_infos = serializers.SerializerMethodField()
|
|
|
|
def get_certification_infos(self, obj):
|
|
result = []
|
|
for c_info in models.CertificationInfo.objects.filter(user_id=obj.user_id).all():
|
|
result.append({
|
|
'name': get_choices_name_from_key(models.CertificationInfo.type_choices, c_info.type),
|
|
'certification_url': get_download_url_from_context(self, obj, '', c_info.certification_url)
|
|
})
|
|
return result
|
|
|
|
def update(self, instance, validated_data):
|
|
return super(AdminUserCertificationSerializer, self).update(instance, validated_data)
|
|
|
|
|
|
class AdminThirdWxSerializer(ThirdWxSerializer):
|
|
class Meta:
|
|
model = models.ThirdWeChatUserInfo
|
|
fields = "__all__"
|
|
read_only_fields = ["id", "openid", "user_id", "head_img_url"]
|
|
|
|
|
|
class AdminDomainNameSerializer(DomainNameSerializer):
|
|
class Meta:
|
|
model = models.UserDomainInfo
|
|
fields = "__all__"
|
|
read_only_fields = ["id", "app_id", "domain_type", "created_time", "cname_id"]
|
|
|
|
domain_type_choices = serializers.SerializerMethodField()
|
|
|
|
def get_domain_type_choices(self, obj):
|
|
return get_choices_dict(obj.domain_type_choices)
|
|
|
|
|
|
class AdminAppReportSerializer(AppReportSerializer):
|
|
class Meta:
|
|
model = models.AppReportInfo
|
|
fields = '__all__'
|
|
read_only_fields = ["id", "app_id", "username", "created_time", "email", "report_reason", "report_type",
|
|
"remote_addr", "bundle_id", "app_name"]
|
|
|
|
report_type_choices = serializers.SerializerMethodField()
|
|
|
|
def get_report_type_choices(self, obj):
|
|
return get_choices_dict(obj.report_type_choices)
|
|
|
|
status_choices = serializers.SerializerMethodField()
|
|
|
|
def get_status_choices(self, obj):
|
|
return get_choices_dict(obj.status_choices)
|
|
|