azfilebackup/azfilebak/timing.py

# coding=utf-8

# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
# --------------------------------------------------------------------------

""" Timing module."""

import time
import datetime

class Timing(object):
    """Timing class."""
    time_format = "%Y%m%d_%H%M%S"

    @staticmethod
    def now_localtime():
        """Return formatted localtime."""
        return time.strftime(Timing.time_format, time.localtime())

    @staticmethod
    def parse(time_str):
        """Parse time string."""
        return time.strptime(time_str, Timing.time_format)

    @staticmethod
    def time_diff(str1, str2):
        """Calculate time difference."""
        t1 = Timing.parse(str1)
        dt1 = datetime.datetime(year=t1.tm_year, month=t1.tm_mon, day=t1.tm_mday, hour=t1.tm_hour, minute=t1.tm_min, second=t1.tm_sec)
        t2 = Timing.parse(str2)
        dt2 = datetime.datetime(year=t2.tm_year, month=t2.tm_mon, day=t2.tm_mday, hour=t2.tm_hour, minute=t2.tm_min, second=t2.tm_sec)
        return dt2 - dt1

    @staticmethod
    def sort(times, selector=lambda x: x):
        """Sort by time."""
        def sort_cmp(a, b):
            """Sort by time diff, and secondary by stripe index when sorting full records."""
            if isinstance(a, type({})) and isinstance(b, type({})):
                return Timing.time_diff_in_seconds(
                    selector(b), selector(a)) or cmp(a.get('stripe_index'), b.get('stripe_index'))
            return Timing.time_diff_in_seconds(selector(b), selector(a))
        return sorted(times, cmp=sort_cmp)

    @staticmethod
    def time_diff_in_seconds(timestr_1, timestr_2):
        """
        >>> Timing.time_diff_in_seconds("20180106_120000", "20180106_120010")
        10
        >>> Timing.time_diff_in_seconds("20180106_110000", "20180106_120010")
        3610
        """
        return int(Timing.time_diff(timestr_1, timestr_2).total_seconds())