Save work.

Author: programmerPhysicist

source/one_way_sync.py

@@ -67,19 +67,9 @@ def sync_todoist_to_habitica():
     todoist_tasks, todo_api = get_tasks(todo_token) # todoist_tasks used to be tod_tasks
 
     tod_tasks = []
-    tzone = None
     for task in todoist_tasks:
         tod_tasks.append(TodTask(task))
 
-    if tzone is None:
-        # assumption is that timezone from Todoist
-        # is the same as local timezone
-        tzone = pytz.timezone(str(get_localzone()))
-
-        for task in tod_tasks:
-            if task.due != '':
-                task.task_dict['due'] = task.due.astimezone(tzone)
-
     # TODO: add back to filter out repeating older than a certain amount?
     # date stuff
     # today = datetime.now()

source/todo_task.py

@@ -10,6 +10,7 @@
 from builtins import *
 from datetime import datetime
 from tzlocal import get_localzone
+import pytz
 
 
 #from .dates import parse_date_utc
@@ -38,6 +39,9 @@ def __init__(self, task=None):
 
         self.__task_dict = task_dict
 
+        tzone = pytz.timezone(str(get_localzone()))
+        self.__task_dict['due']['date'] = self.due.astimezone(tzone)
+
     @property
     #Get the task dictionary as is
     def task_dict(self):

test/integration/oneWaySyncTests/dailies_test.py

@@ -0,0 +1,183 @@
+# pylint: disable=missing-function-docstring, missing-module-docstring, invalid-name, import-error
+# integration test for one_way_sync.py
+import pickle
+from datetime import datetime
+from dateutil.tz import tzoffset
+import pytest
+import requests
+import vcr
+from mockito import when, mock, when2, verify, captor, ANY, arg_that
+from one_way_sync import sync_todoist_to_habitica
+from todoist_api_python import models
+from common_fixtures import empty_pickle, fake_config_file, mock_web_calls # pylint: disable=unused-import
+# pylint: enable=invalid-name
+
+
+def case1():
+    hab_val = {"data": []}
+
+    due_dict = {'date': '2024-12-27',
+                'datetime': None,
+                'is_recurring': True,
+                'string': 'every day',
+                'timezone': None}
+    duedate = models.Due.from_dict(due_dict)
+    created_at = '2025-01-04T00:00:00.0Z'
+
+    todoist_task = models.Task(None,  # assignee id
+                               None,  # assigner id
+                               0,     # comment count
+                               False, # is_completed
+                               'Test task 1', # content
+                               created_at,    # created_at
+                               '59292300',    # creater_id
+                               '',            # description
+                               duedate,       # due
+                               '8296278113',  # id
+                               [],            # labels
+                               0,             # order
+                               None,          # parent id
+                               1,             # priority
+                               '9187482462',  # project id
+                               '19099659',    # section id
+                               None,          # url
+                               ''             # duration
+                               )
+    todo_tasks = [todoist_task]
+    completed_todos = []
+
+    inputs = {'hab_task': hab_val,
+              'completed_habs': hab_val,
+              'todo_tasks': todo_tasks,
+              'done_tasks': completed_todos}
+
+    return inputs
+
+
+'''def check_headers(headers):
+    assert headers['url'] == 'https://habitica.com'
+    assert headers['x-api-user'] == 'cd18fc9f-b649-4384-932a-f3bda6fe8102'
+    assert headers['x-api-key'] == '18f22441-2c87-6d8e-fb2a-3fa670837b5a'
+'''
+
+
+def verify_post_request(data):
+    if data['text'] == 'Test task 1':
+        assert data['type'] == 'todo'
+        assert data['text'] == 'Test task 1'
+        assert data['date'] == '12/27/2024, 00:00:00'
+        assert data['alias'] == '8296278113'
+        assert data['priority'] == '2'
+        assert data['attribute'] == 'str'
+        return True
+    return False
+
+'''
+def verify_put_request(the_url, the_data):
+    assert the_url.value == 'https://habitica.com/api/v3/tasks/96935939'
+    data = the_data.value
+    assert data['alias'] == '96935939'
+    assert data['text'] == 'Some test task'
+    assert data['priority'] == 1
+'''
+
+
+def verify_pickle_dump(dump_dict):
+    data = dump_dict.value
+    # check 'simple' values
+    assert '8296278113' in data.keys()
+    data = data['8296278113']
+    assert data['recurs'] == 'No'
+    assert data['duelast'] == 'NA'
+    # Get objects to verify
+    assert 'tod' in data.keys()
+    tod_task = data['tod']
+    assert 'hab' in data.keys()
+    hab_task = data['hab']
+    # Check tod_task
+    tod_data = tod_task.task_dict
+    assert tod_data['content'] == 'Test task 1'
+    assert tod_data['id'] == '8296278113'
+    assert tod_data['created_at'] == '2025-01-04T00:00:00.0Z'
+    assert tod_data['priority'] == 1
+    # Check hab_task
+    hab_data = hab_task.task_dict
+    assert hab_data['type'] == 'daily'
+    assert hab_data['alias'] == '96935939'
+    assert hab_data['text'] == 'Some test task'
+    assert hab_data['priority'] == 1
+    expected_due = datetime(2024, 12, 27, tzinfo=tzoffset(None, -25200))
+    due = hab_task.due
+    assert due == expected_due
+
+
+# pylint: disable=missing-class-docstring
+class TestDailies:
+    test_vcr = vcr.VCR(
+        serializer='yaml',
+        cassette_library_dir="/tmp/throwaway",
+        record_mode='none'
+    )
+
+    # pylint: disable=redefined-outer-name, unused-argument, too-many-locals, too-few-public-methods
+    @pytest.mark.parametrize("mock_web_calls", [case1()], indirect=True)
+    @pytest.mark.parametrize("pickle_in", [empty_pickle()], indirect=True)
+    def test(self,
+             fake_config_file,
+             mock_web_calls):
+        # pylint: enable=redefined-outer-name, unused-argument
+
+        # set default response
+        response = mock({'status': 200, 'ok': True}, spec=requests.Response)
+
+        # mock out post to Habitica
+        when(requests).post(...).thenReturn(response)
+
+        # mock out put to Habitica
+        when(requests).put(...).thenReturn(response)
+
+        # mock out web call to get id
+        hab_task = {'text': 'Some test task', 'priority': '', 'attribute': '',
+                    'type': 'todo', '_id': 'a94e8f46-5c14-f14a-f189-e669e239730a',
+                    'completed': False, 'alias': '96935939', 'date': '12/27/2024, 00:00:00'}
+        hab_val2 = {"data": hab_task}
+
+        response2 = mock({'status': 200, 'ok': True}, spec=requests.Response)
+        task_url = 'https://habitica.com/api/v3/tasks/8296278113'
+        when(requests).get(headers={'url': 'https://habitica.com',
+                                    'x-api-user': 'cd18fc9f-b649-4384-932a-f3bda6fe8102',
+                                    'x-api-key': '18f22441-2c87-6d8e-fb2a-3fa670837b5a'},
+                           url=task_url).thenReturn(response2)
+        when(response2).json().thenReturn(hab_val2)
+
+        # mock dump of pickle file
+        pkl_out = mock()
+        pkl_file = mock()
+        when2(open, ...).thenCallOriginalImplementation()
+        when2(open, 'oneWay_matchDict.pkl', 'wb').thenReturn(pkl_file)
+        when(pickle).Pickler(...).thenReturn(pkl_out)
+        when(pkl_out).dump(...)
+
+        # using get_all_habtasks() which contains requests.get(), uses the monkeypatch
+        with self.test_vcr.use_cassette("null.yaml"):
+            sync_todoist_to_habitica()
+
+        # verify post request
+        '''the_url = captor(ANY(str))
+        the_data = captor(ANY(dict))
+        the_headers = captor(ANY(dict))
+        verify(requests, times=1).post(url='https://habitica.com/api/v3/tasks/user/',
+                                       data=arg_that(verify_post_request),
+                                       headers=the_headers)
+        # check_headers(the_headers.value)'''
+
+        # verify put request
+        '''the_url = captor(ANY(str))
+        verify(requests, times=1).put(headers=the_headers, url=the_url, data=the_data)'''
+        # check_headers(the_headers.value)
+        # verify_put_request(the_url, the_data)
+
+        # verify pickle dump
+        dump_dict = captor(ANY(dict))
+        verify(pkl_out, times=1).dump(dump_dict)
+        verify_pickle_dump(dump_dict)

test/integration/oneWaySyncTests/end_to_end_integration_test.py

@@ -157,6 +157,25 @@ def test_end_to_end(self,
             save_pickle_for_test(data)
             assert len(data.keys()) == expected_vals.keys
 
+            num_todos = 0
+            num_habits = 0
+            num_dailies = 0
+            num_other = 0
+            for value in data.values():
+                hab_type = value['hab'].category
+                if hab_type == 'todo':
+                    num_todos += 1
+                elif hab_type == 'habit':
+                    num_habits += 1
+                elif hab_type == 'daily':
+                    num_dailies += 1
+                else:
+                    num_other += 1
+            assert num_todos == 77
+            assert num_habits == 0
+            assert num_dailies < 0
+            assert num_other == 0
+
             # check put
             if expected_vals.iters != 0:
                 the_url = captor(ANY(str))