# -*- coding: utf-8 -*-

import datetime
import time


def str_datetime_to_timestamp(str_datetime, formate="%Y-%m-%d %H:%M:%S"):
    """将字符串的时间转换为时间戳"""

    # 先装换成struct_time对象
    time_array = time.strptime(str_datetime, formate)

    # 将struct_time装换成时间戳
    time_stamp = int(time.mktime(time_array))
    return time_stamp


def format_datetime(str_datetime, old_format, new_format):
    """字符串时间格式更改，如a = "2013-10-10 23:40:00",想改为 a = "2013/10/10 23:40:00"""
    time_array = time.strptime(str_datetime, old_format)
    new_format = time.strftime(new_format, time_array)

    return new_format


def timestamp_to_format_time(timestamp, formate):
    """时间戳转换为指定格式日期"""
    time_array = time.localtime(timestamp)
    format_time = time.strftime(formate, time_array)

    return format_time


def get_now_format(define_formate, type=0):
    """获取当前时间并装换指定格式"""
    if type:
        # 方式1：
        now = datetime.datetime.now()
        return now.strftime(define_formate)
    else:
        # 方式2
        now = int(time.time())  # 时间戳
        time_array = time.localtime(now)
        return time.strftime(define_formate, time_array)


def get_datetime_by_days(day, define_formate):
    """获取几天前或几天后的日期时间"""
    t = datetime.datetime.now() - datetime.timedelta(days=day)
    time_stamp = int(time.mktime(t.timetuple()))

    return t.strftime(define_formate), time_stamp


def get_datetime_obj(str_datetime, define_format):
    """日期时间按字符串装换为datetime对象"""
    t_obj = datetime.datetime.strptime(str_datetime, define_format)

    return t_obj