#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2020-02-10 23:12
# @Author  : Lemon
# @File    : common.py
# @Software: PyCharm

import datetime
import demjson
import base64
from enum import Enum
from attr import attrs, attrib


@attrs(cmp=False)
class WorkPlace():
    employer_name = attrib("")
    employer_id = attrib("")
    position_text = attrib("")
    position_id = attrib("")
    location_name = attrib("")
    location_id = attrib("")
    description = attrib("")
    current = attrib(False, type=bool)
    start_time = attrib(None, type=datetime.date)
    end_time = attrib(None, type=datetime.date)

    def to_dict(self):
        data = {
            'employer_name': self.employer_name,
            'employer_id': self.employer_id,
            'position_text': self.position_text,
            'position_id': self.position_id,
            'location_name': self.location_name,
            'location_id': self.location_id,
            'description': self.description,
            'date_start[year]': str(self.start_time.year),
            'date_start[month]': str(self.start_time.month),
            'date_start[day]': str(self.start_time.day),
            'date_end[year]': str(self.end_time.year) if self.end_time else '',
            'date_end[month]': str(self.end_time.month) if self.end_time else '',
            'date_end[day]': str(self.end_time.day) if self.end_time else '',
        }
        if self.current:
            data['date[current]'] = 'on'
        return data

    @classmethod
    def from_dict(cls, data: dict):
        self = cls()
        for k, v in data.items():
            if hasattr(self, k):
                setattr(self, k, v)

        start_time = []
        end_time = []
        for x in ['year', 'month', 'day']:
            if data.get(f'date_start[{x}]'):
                start_time.append(int(data.get(f'date_start[{x}]')))

            if data.get(f'date_end[{x}]'):
                end_time.append(int(data.get(f'date_end[{x}]')))
        if start_time:
            self.start_time = datetime.date(*start_time)

        if end_time:
            self.end_time = datetime.date(*end_time)
        else:
            self.current = True

        return self


@attrs(cmp=False)
class College():
    school_text = attrib("")
    school_id = attrib("")
    subject = attrib([], type=list, converter=lambda x: [] if x is None else x)
    school_type = attrib("college")
    description = attrib("")
    graduated = attrib(False, type=bool)  # 'graduated': 'on',
    start_time = attrib(None, type=datetime.date)
    end_time = attrib(None, type=datetime.date)

    def to_dict(self):
        data = {
            'school_text': self.school_text,
            'school_id': self.school_id,
            'date_start[year]': str(self.start_time.year),
            'date_start[month]': str(self.start_time.month),
            'date_start[day]': str(self.start_time.day),
            'date_end[year]': str(self.end_time.year) if self.end_time else '',
            'date_end[month]': str(self.end_time.month) if self.end_time else '',
            'date_end[day]': str(self.end_time.day) if self.end_time else '',
            'description': self.description,
            'school_type': self.school_type
        }
        if self.graduated:
            data['graduated'] = 'on'
        if self.subject:
            for i, sub in enumerate(self.subject):
                assert len(sub) == 2, '格式不对，要求格式（concentration_text，concentration_ids）'
                data["concentration_text[%d]" % i] = sub[0]
                data["concentration_ids[%d]" % i] = sub[1]
        return data

    @classmethod
    def from_dict(cls, data: dict):
        self = cls()
        for k, v in data.items():
            if hasattr(self, k):
                setattr(self, k, v)

        start_time = []
        end_time = []
        for x in ['year', 'month', 'day']:
            if data.get(f'date_start[{x}]'):
                start_time.append(int(data.get(f'date_start[{x}]')))

            if data.get(f'date_end[{x}]'):
                end_time.append(int(data.get(f'date_end[{x}]')))
        if start_time:
            self.start_time = datetime.date(*start_time)

        if end_time:
            self.end_time = datetime.date(*end_time)
            self.graduated = True

        return self


class TaskStatus():
    EXCEPTION = -2
    FAILED = -1
    EXECUTE = 0
    SUCCESS = 1


class Exchange(Enum):
    CLIENT = 0

    # SERVER = 1
    # NSQ = 2
    # MQTT = 3

    @classmethod
    def get(cls, code) -> Enum:
        if isinstance(code, str):
            code = int(code)

        keys = [x for x in dir(cls) if not x.startswith('__')]
        for key in keys:
            member = getattr(cls, key)
            if member.value == code:
                return member
        return None


def todict(obj, include: list = None):
    keys = dir(obj)
    res = {}
    if include:
        for key in include:
            try:
                value = getattr(obj, key)
                res[key] = value
            except:
                pass
        return res
    else:
        for key in keys:
            if not key.startswith('_'):
                value = getattr(obj, key)
                res[key] = value
        return res


def tobase64(obj) -> str:
    if isinstance(obj, (dict, list)):
        obj = demjson.encode(obj)
    if isinstance(obj, str):
        obj = obj.encode('utf-8')
    if isinstance(obj, bytes):
        bin = base64.b64encode(obj)
        return str(bin, 'utf-8')
    raise BaseException("must str,list,dict,bytes")


def frombase64(base) -> dict:
    string = base64.b64decode(base).decode()
    return demjson.decode(string)


def format_proxies(conf: dict, format='requests'):
    '''
    e.g.
    {
     'type': 'socks5',
     'host': '47.56.152.111',
     'pass': 'nantian888',
     'port': '1080',
     'user': 'ntkj'
    }
    '''
    type_ = conf.get('type', None)
    host = conf.get('host', None)
    port = conf.get('port', None)
    user = conf.get('user', None)
    pass_ = conf.get('pass', None)

    if format == 'requests':
        info = []
        info.append(type_)
        info.append("://")
        p_auth = []
        if user:
            p_auth.append(str(user))
            p_auth.append(":")
        if pass_:
            p_auth.append(str(pass_))
        if p_auth:
            info.append(''.join(p_auth))
            info.append('@')
        info.append(host)
        if port:
            info.append(':')
            info.append(str(port))
        text = ''.join(info)
        return {'http': text, 'https': text}
    else:
        _type_ = getattr(__import__('socks'), 'PROXY_TYPES').get(type_.upper())
        assert _type_, '代理协议错误，可选socks5，socks4，http'
        proxy = {
            "proxy_type": _type_,
            "proxy_addr": host,
            "proxy_port": int(port),
        }
        if user:
            proxy['proxy_username'] = user
        if pass_:
            proxy['proxy_password'] = pass_
        return proxy


if __name__ == '__main__':
    conf = {'type': 'http',
            'host': '47.56.152.111',
            'pass': 'nantian888',
            'port': '1080',
            'user': 'ntkj'}
    print(format_proxies(conf, '1'))