Add an option to start the timer 1 second after the last time unit

Author: dvklopfenstein

timetracker/cmd/start.py

@@ -3,6 +3,8 @@
 __copyright__ = 'Copyright (C) 2025-present, DV Klopfenstein, PhD. All rights reserved.'
 __author__ = "DV Klopfenstein, PhD"
 
+import os.path as op
+from timetracker.csvfile import CsvFile
 from timetracker.epoch.epoch import get_dt_at
 from timetracker.cmd import common
 
@@ -13,7 +15,8 @@ def cli_run_start(fnamecfg, args):
         fnamecfg,
         args.name,
         start_at=args.at,
-        force=args.force)
+        force=args.force,
+        uname=args.name)
 
 def _run_start(fnamecfg, name=None, start_at=None, **kwargs):
     """Initialize timetracking on a project"""
@@ -36,8 +39,13 @@ def run_start(cfgproj, name=None, start_at=None, **kwargs):
         common.prt_elapsed(startobj)
 
     # Set (if not started) or reset (if start is forced) starting time
-    if not startobj.started() or force:
-        starttime = get_dt_at(start_at, kwargs.get('now'), kwargs.get('defaultdt'))
+    if not startobj.started() or force or start_at == 'last':
+        starttime = _get_starttime(start_at, cfgproj, kwargs)
+           # .get('now'),
+           # kwargs.get('defaultdt'),
+           # kwargs.get('dirhome'),
+           # kwargs.get(')
+        print(f'STARTTIME: {starttime}')
         if starttime is None:
             return startobj
         startobj.wr_starttime(starttime, kwargs.get('activity'), kwargs.get('tag'))
@@ -53,6 +61,17 @@ def run_start(cfgproj, name=None, start_at=None, **kwargs):
             print(f'Run `trk start --at {start_at} --force` to force restart')
     return startobj
 
+
+def _get_starttime(start_at, cfgproj, kwargs):
+    if start_at != 'last':
+        return get_dt_at(start_at, kwargs.get('now'), kwargs.get('defaultdt'))
+    fcsv = cfgproj.get_filename_csv(kwargs.get('uname'), kwargs.get('dirhome'))
+    if op.exists(fcsv):
+        csvfile = CsvFile(fcsv)
+        next_startdatetime = csvfile.get_next_start_datetime()
+        return next_startdatetime if next_startdatetime else None
+    return get_dt_at(None, kwargs.get('now'), kwargs.get('defaultdt'))
+
 def _get_msg(start_at, force):
     if force:
         return "start reset to"

timetracker/csvfile.py

@@ -4,11 +4,12 @@
 __copyright__ = 'Copyright (C) 2025-present, DV Klopfenstein, PhD. All rights reserved.'
 __author__ = "DV Klopfenstein, PhD"
 
+import os
 from os.path import exists
 from collections import namedtuple
 from datetime import timedelta
 # https://docs.python.org/3/library/csv.html
-from csv import writer
+import csv
 
 from timetracker.csvutils import get_hdr_itr
 from timetracker.epoch.calc import dt_from_str
@@ -47,6 +48,7 @@ def wr_csvline(self, dta, delta, csvfields):
             with open(self.fcsv, 'w', encoding='utf8') as csvfile:
                 self.wr_hdrs(csvfile)
         # Print time information into csv
+        _writer = csv.writer
         with open(self.fcsv, 'a', encoding='utf8') as csvfile:
             # timedelta(days=0, seconds=0, microseconds=0,
             #           milliseconds=0, minutes=0, hours=0, weeks=0)
@@ -55,20 +57,25 @@ def wr_csvline(self, dta, delta, csvfields):
             data = [str(dta),
                     str(delta),
                     csvfields.activity, csvfields.message, csvfields.tags]
-            writer(csvfile, lineterminator='\n').writerow(data)
+            _writer(csvfile, lineterminator='\n').writerow(data)
             return data
         return None
 
     def wr_hdrs(self, prt):
         """Write header"""
         print(','.join(self.hdrs), file=prt)
 
+    def get_next_start_datetime(self):
+        """Get the next start_datetime after the last line in a timetracker csv file"""
+        ntd = self._rd_last_line()
+        return ntd.start_datetime + ntd.duration + timedelta(seconds=1) if ntd else None
+
     # ------------------------------------------------------------------
     def _get_ntdata(self, csvlines):
         """Get data where start and stop are datetimes; timdelta is calculated from them"""
         nto = self.nto
         def _get_nt(row):
-            assert len(row) == 5, f'{self.fcsv} ROW[{len(row)}]: {row}'
+            assert len(row) == 5, row
             return nto(
                 start_datetime=dt_from_str(row[0]),
                 duration=td_from_str(row[1]),
@@ -103,5 +110,28 @@ def _read_csv(self, fnc_csvlines):
                 return self.ntrdcsv(results=fnc_csvlines(itr), error=error)
         return self.ntrdcsv(results=None, error=error)
 
+    def _rd_last_line(self):
+        """Return the last line in the csv file"""
+        try:
+            fptr = open(self.fcsv, mode='rb')  #encoding='utf8')
+        except (PermissionError, OSError) as err:
+            print(type(err).__name__, err.args)
+        else:
+            with fptr as csvstrm:
+                try:
+                    csvstrm.seek(-2, os.SEEK_END)
+                    while csvstrm.read(1) != b'\n':
+                        csvstrm.seek(-2, os.SEEK_CUR)
+                except OSError:
+                    csvstrm.seek(0)
+                csv_line   = csvstrm.readline().decode()
+                csv_reader = csv.reader([csv_line])
+                row = next(csv_reader)
+                nts = self._get_ntdata([row])
+                if nts:
+                    assert len(nts) == 1, nts
+                    return nts[0]
+        return None
+
 
 # Copyright (C) 2025-present, DV Klopfenstein, PhD. All rights reserved.