ASJPush/Service/CommonService.py

# -*- coding: utf-8 -*-
import base64
import datetime
import os
import time
import hashlib
from pathlib import Path
from random import Random
import ipdb
import simplejson as json
from boto3 import Session
from django.core import serializers
from django.utils import timezone
from pyipip import IPIPDatabase

from AnsjerPush.config import BASE_DIR, ACCESS_KEY_ID, SECRET_ACCESS_KEY, REGION_NAME, PUSH_BUCKET, SYS_EVENT_TYPE_LIST, \
    DATA_PUSH_EVENT_TYPE_LIST
from Object.enums.EventTypeEnum import EventTypeEnumObj


# 复用性且公用较高封装代码在这
class CommonService:

    # 添加模糊搜索
    @staticmethod
    def get_kwargs(data={}):
        kwargs = {}
        for (k, v) in data.items():
            if v is not None and v != u'':
                kwargs[k + '__icontains'] = v
        return kwargs

    # 定义静态方法
    # 格式化query_set转dict
    @staticmethod
    def qs_to_dict(query_set):
        sqlJSON = serializers.serialize('json', query_set)
        sqlList = json.loads(sqlJSON)
        sqlDict = dict(zip(["datas"], [sqlList]))
        return sqlDict

    # 获取文件大小
    @staticmethod
    def get_file_size(file_path='', suffix_type='', decimal_point=0):

        # for x in ['bytes', 'KB', 'MB', 'GB', 'TB']:
        # path = Path() / 'D:/TestServer/123444.mp4'
        path = Path() / file_path
        size = path.stat().st_size
        mb_size = 0.0
        if suffix_type == 'MB':
            mb_size = size / 1024.0 / 1024.0
        if decimal_point != 0:
            mb_size = round(mb_size, decimal_point)
        return mb_size

    @staticmethod
    def get_param_flag(data=[]):
        # print(data)
        flag = True
        for v in data:
            if v is None:
                flag = False
                break
        return flag

    @staticmethod
    def get_ip_address(request):
        """
        获取ip地址
        :param request:
        :return:
        """
        try:
            real_ip = request.META['HTTP_X_FORWARDED_FOR']
            clientIP = real_ip.split(",")[0]
        except:
            try:
                clientIP = request.META['REMOTE_ADDR']
            except Exception as e:
                clientIP = ''
        return clientIP

    # @获取一天每个小时的datetime.datetime
    @staticmethod
    def getTimeDict(times):
        time_dict = {}
        t = 0
        for x in range(24):
            if x < 10:
                x = '0' + str(x)
            else:
                x = str(x)
            a = times.strftime("%Y-%m-%d") + " " + x + ":00:00"
            time_dict[t] = timezone.datetime.strptime(a, '%Y-%m-%d %H:%M:%S')
            t += 1
        return time_dict

    # 根据ip获取地址
    @staticmethod
    def getAddr(ip):
        base_dir = BASE_DIR
        # ip数据库
        db = IPIPDatabase(base_dir + '/DB/17monipdb.dat')
        addr = db.lookup(ip)
        ts = addr.split('\t')[0]
        return ts

    # 通过ip检索ipip指定信息 lang为CN或EN
    @staticmethod
    def getIpIpInfo(ip, lang, update=False):
        ipbd_dir = BASE_DIR + "/DB/mydata4vipday2.ipdb"
        db = ipdb.City(ipbd_dir)
        if update:
            rr = db.reload(ipbd_dir)
        info = db.find_map(ip, lang)
        return info

    @staticmethod
    def getUserID(userPhone='13800138000', getUser=True, setOTAID=False, μs=True):
        if μs == True:
            if getUser == True:
                timeID = str(round(time.time() * 1000000))
                userID = timeID + userPhone

                return userID
            else:
                if setOTAID == False:
                    timeID = str(round(time.time() * 1000000))
                    ID = userPhone + timeID

                    return ID
                else:
                    timeID = str(round(time.time() * 1000000))
                    eID = '13800' + timeID + '138000'
                    return eID
        else:
            if getUser == True:
                timeID = str(round(time.time() * 1000))
                userID = timeID + userPhone

                return userID
            else:
                if setOTAID == False:
                    timeID = str(round(time.time() * 1000))
                    ID = userPhone + timeID

                    return ID
                else:
                    timeID = str(round(time.time() * 1000))
                    eID = '13800' + timeID + '138000'
                    return eID

    # 生成随机数
    @staticmethod
    def RandomStr(randomlength=8, number=True):
        str = ''
        if number == False:
            characterSet = 'AaBbCcDdEeFfGgHhIiJjKkLlMmNnOoPpQqRrSsT' \
                           'tUuVvWwXxYyZz0123456789'
        else:
            characterSet = '0123456789'

        length = len(characterSet) - 1

        random = Random()
        for index in range(randomlength):
            str += characterSet[random.randint(0, length)]
        return str

    # 生成订单好
    @staticmethod
    def createOrderID():
        random_id = CommonService.RandomStr(6, True)
        order_id = datetime.datetime.now().strftime('%Y%m%d%H%M%S') + str(random_id)
        print('orderID:')
        print(order_id)
        return order_id

    # qs转换list datetime处理
    @staticmethod
    def qs_to_list(qs):
        res = []
        # print(qs)
        for ps in qs:
            if 'add_time' in ps:
                ps['add_time'] = ps['add_time'].strftime("%Y-%m-%d %H:%M:%S")
            if 'update_time' in ps:
                ps['update_time'] = ps['update_time'].strftime("%Y-%m-%d %H:%M:%S")
            if 'end_time' in ps:
                ps['end_time'] = ps['end_time'].strftime("%Y-%m-%d %H:%M:%S")
            if 'data_joined' in ps:
                if ps['data_joined']:
                    ps['data_joined'] = ps['data_joined'].strftime("%Y-%m-%d %H:%M:%S")
                else:
                    ps['data_joined'] = ''
            res.append(ps)
        return res

    # 获取当前时间
    @staticmethod
    def get_now_time_str(n_time, tz, lang):
        print(n_time)
        print(tz)
        print(lang)
        try:
            tz = tz.replace(':', '.')
            n_time = int(n_time) + 3600 * float(tz)
        except:
            n_time = int(n_time)
        if lang == 'cn':
            return time.strftime('%Y-%m-%d %H:%M:%S', time.gmtime(int(n_time)))
        else:
            return time.strftime('%m-%d-%Y %H:%M:%S', time.gmtime(int(n_time)))

    @staticmethod
    def app_log_log(uid='None', tz='0'):
        file_path = '/'.join((BASE_DIR, 'static/app_log.log'))
        file = open(file_path, 'a+')
        file.write("uid:" + uid + "; " + "; tz:" + tz)
        file.write('\n')
        file.flush()
        file.close()

    @classmethod
    def upload_images(cls, file_dict, dir_path):
        """
        上传图片到S3,并删除本地图片
        @param file_dict: S3图片路径
        @param dir_path: 本地图片路径
        @return: boolean
        """
        try:
            s3 = Session(
                aws_access_key_id=ACCESS_KEY_ID,
                aws_secret_access_key=SECRET_ACCESS_KEY,
                region_name=REGION_NAME
            ).resource('s3')

            for file_path, upload_path in file_dict.items():
                upload_data = open(file_path, 'rb')
                s3.Bucket(PUSH_BUCKET).put_object(Key=upload_path, Body=upload_data)

            # 删除图片
            cls.del_path(dir_path)
            cls.del_path(dir_path + '.jpg')
            return True
        except Exception as e:
            print(repr(e))
            return False

    @classmethod
    def del_path(cls, path):
        """
        删除目录文件
        @param path: 文件路径
        @return: None
        """
        if not os.path.exists(path):
            return
        if os.path.isfile(path):
            os.remove(path)
        else:
            items = os.listdir(path)
            for f in items:
                c_path = os.path.join(path, f)
                if os.path.isdir(c_path):
                    cls.del_path(c_path)
                else:
                    os.remove(c_path)
            os.rmdir(path)

    @staticmethod
    def getMD5Sign(data, key):
        '''
        魅族MD5签名
        '''
        dataList = []
        for k in sorted(data):
            dataList.append("%s=%s" % (k, data[k]))
        data = (''.join(dataList))
        data = data + key
        sign = hashlib.md5(data.encode(encoding="utf-8")).hexdigest()
        return sign

    @staticmethod
    def check_time_stamp_token(token, time_stamp):
        # 时间戳token校验
        if not all([token, time_stamp]):
            return False
        try:
            token = int(CommonService.decode_data(token))
            time_stamp = int(time_stamp)
            now_time = int(time.time())
            distance = now_time - time_stamp
            if token != time_stamp or distance > 60000 or distance < -60000:  # 为了全球化时间控制在一天内
                return False
            return True
        except Exception as e:
            print(e)
            return False

    @staticmethod
    def decode_data(content, start=1, end=4):
        """
        数据解密
        @param content: 数据内容
        @param start: 起始长度
        @param end: 结束长度
        @return content: 解密的数据
        """
        if not content:
            return ''
        for i in range(start, end):
            content = base64.b64decode(content)
            content = content.decode('utf-8')
            content = content[i:-i]
        return content

    @staticmethod
    def timestamp_to_str(timestamp):
        """
        时间戳转时间字符串
        @param timestamp: 时间戳
        @return time_str: 时间字符串
        """
        struct_time = time.localtime(timestamp)
        time_str = time.strftime("%Y-%m-%d %H:%M:%S", struct_time)
        return time_str

    @staticmethod
    def get_jump_type(event_type):
        """
        获取跳转类型
        @param event_type: 事件类型
        @return event_type: 跳转类型,1:推送消息,2:系统消息,3:音视频通话消息
        """
        # 跳转类型,1:推送消息,2:系统消息,3:音视频通话消息
        jump_type = 1
        event_type = int(event_type)
        if event_type in SYS_EVENT_TYPE_LIST:
            jump_type = 2
        elif event_type in DATA_PUSH_EVENT_TYPE_LIST:
            jump_type = 3
        return jump_type