【Python】cronを生成するモジュールを作った

1
import datetime
2

3
def generate_cron_from_datetime_now(minutes_scheduled_later : int, time_difference : int = 0) -> str:
4
    """
5
    e.g.
6
    '0 19 * * *'  # At 04:00. – https://crontab.guru"
7
    """
8
    if type(minutes_scheduled_later) != int:
9
        raise TypeError("TypeError: minutes_scheduled_later must be int type.")
10
    if type(time_difference) != int:
11
        raise TypeError("TypeError: time_difference must be int type.")
12
    cron_minute = 0
13
    cron_hour = 0
14
    dt_now = datetime.datetime.now(
15
        datetime.timezone(datetime.timedelta(hours=time_difference))
16
    )
17
    minutes_from_an_hour = 60
18
    if dt_now.minute + minutes_scheduled_later >= minutes_from_an_hour:
19
        cron_minute = dt_now.minute + minutes_scheduled_later - minutes_from_an_hour
20
        cron_hour = dt_now.hour + 1
21
    else:
22
        cron_minute = dt_now.minute + minutes_scheduled_later
23
        cron_hour = dt_now.hour
24
    hours_from_a_day = 24
25
    if cron_hour >= hours_from_a_day:
26
        cron_hour = cron_hour - hours_from_a_day
27

28
    cron = f"{cron_minute} {cron_hour} * * *"
29
    return cron

1
import pytest
2

3
from src.landmasterlibrary.generaltool import generate_cron_from_datetime_now
4

5
class Test_Generaltool:
6

7
    @pytest.mark.freeze_time("2022-01-12 13:23:43")
8
    def test_generate_cron_from_datetime_now_1_1(self):
9
        minutes_scheduled_later = 10
10
        time_difference = 0
11
        actual = generate_cron_from_datetime_now(minutes_scheduled_later, time_difference)
12
        expected_hours = 13
13
        expected_minutes = 33
14
        expected = f"{expected_minutes} {expected_hours} * * *"
15
        assert actual == expected
16

17
    @pytest.mark.freeze_time("2022-01-12 13:53:43")
18
    def test_generate_cron_from_datetime_now_1_2(self):
19
        minutes_scheduled_later = 10
20
        time_difference = 0
21
        actual = generate_cron_from_datetime_now(minutes_scheduled_later, time_difference)
22
        expected_hours = 14
23
        expected_minutes = 3
24
        expected = f"{expected_minutes} {expected_hours} * * *"
25
        assert actual == expected
26

27
    @pytest.mark.freeze_time("2022-01-12 23:53:43")
28
    def test_generate_cron_from_datetime_now_1_3(self):
29
        minutes_scheduled_later = 10
30
        time_difference = 0
31
        actual = generate_cron_from_datetime_now(minutes_scheduled_later, time_difference)
32
        expected_hours = 0
33
        expected_minutes = 3
34
        expected = f"{expected_minutes} {expected_hours} * * *"
35
        assert actual == expected
36

37
    @pytest.mark.freeze_time("2022-01-12 13:23:43")
38
    def test_generate_cron_from_datetime_now_2_1(self):
39
        minutes_scheduled_later = 10
40
        time_difference = 2
41
        actual = generate_cron_from_datetime_now(minutes_scheduled_later, time_difference)
42
        expected_hours = 15
43
        expected_minutes = 33
44
        expected = f"{expected_minutes} {expected_hours} * * *"
45
        assert actual == expected
46

47
    @pytest.mark.freeze_time("2022-01-12 13:23:43")
48
    def test_generate_cron_from_datetime_now_2_2(self):
49
        minutes_scheduled_later = 10
50
        time_difference = -2
51
        actual = generate_cron_from_datetime_now(minutes_scheduled_later, time_difference)
52
        expected_hours = 11
53
        expected_minutes = 33
54
        expected = f"{expected_minutes} {expected_hours} * * *"
55
        assert actual == expected
56

57
    @pytest.mark.freeze_time("2022-01-12 13:23:43")
58
    def test_generate_cron_from_datetime_now_3_1(self):
59
        minutes_scheduled_later = 10
60
        actual = generate_cron_from_datetime_now(minutes_scheduled_later)
61
        expected_hours = 13
62
        expected_minutes = 33
63
        expected = f"{expected_minutes} {expected_hours} * * *"
64
        assert actual == expected
65

66
    @pytest.mark.freeze_time("2022-01-12 23:53:43")
67
    def test_generate_cron_from_datetime_now_4_1(self):
68
        minutes_scheduled_later = "10"
69
        time_difference = 0
70
        with pytest.raises(TypeError) as e:
71
            actual = generate_cron_from_datetime_now(minutes_scheduled_later, time_difference)
72

73
    @pytest.mark.freeze_time("2022-01-12 23:53:43")
74
    def test_generate_cron_from_datetime_now_4_2(self):
75
        minutes_scheduled_later = 10
76
        time_difference = "0"
77
        with pytest.raises(TypeError) as e:
78
            actual = generate_cron_from_datetime_now(minutes_scheduled_later, time_difference)
79

80
    @pytest.mark.freeze_time("2022-01-12 23:53:43")
81
    def test_generate_cron_from_datetime_now_4_3(self):
82
        minutes_scheduled_later = None
83
        time_difference = 0
84
        with pytest.raises(TypeError) as e:
85
            actual = generate_cron_from_datetime_now(minutes_scheduled_later, time_difference)
86

87
    @pytest.mark.freeze_time("2022-01-12 23:53:43")
88
    def test_generate_cron_from_datetime_now_4_4(self):
89
        minutes_scheduled_later = 10
90
        time_difference = None
91
        with pytest.raises(TypeError) as e:
92
            actual = generate_cron_from_datetime_now(minutes_scheduled_later, time_difference)
93

94
    @pytest.mark.freeze_time("2022-01-12 23:53:43")
95
    def test_generate_cron_from_datetime_now_5_1(self):
96
        with pytest.raises(TypeError) as e:
97
            actual = generate_cron_from_datetime_now()