wip

hubstaff-cli/src/timesheet.py

@@ -0,0 +1,112 @@
+from services.hubstaff_api import HubstaffAPI
+import os
+import shutil
+import asyncio
+from datetime import datetime, timedelta
+
+
+async def main():
+    # Load the Hubstaff API key from environment variables
+    api_key = os.getenv("HUBSTAFF_API_KEY")
+    if not api_key:
+        print("Error: HUBSTAFF_API_KEY environment variable not set.")
+        return
+
+    # Load the Hubstaff organization ID from environment variables
+    org_id = os.getenv("HUBSTAFF_ORG_ID")
+    if not org_id:
+        print("Error: HUBSTAFF_ORG_ID environment variable not set.")
+        return
+
+    # Initialize the Hubstaff API client
+    hubstaff = HubstaffAPI(refresh_token=api_key)
+
+    try:
+        # Fetch user info
+        user_info = await hubstaff.get_current_user()
+        user_id = user_info["user"]["id"]
+
+        # Define the date range for the timesheet
+        end_date = datetime.now()
+        start_date = end_date - timedelta(days=7)
+
+        # Fetch timesheet data
+        timesheet_data = await hubstaff.get_timesheets(
+            user_id, start=start_date.isoformat(), stop=end_date.isoformat()
+        )
+
+        # Check if timesheet_data is valid and "times" is a list before proceeding
+        times = timesheet_data.get("times") if timesheet_data else None
+        if not times or not isinstance(times, list):
+            print("Error: No timesheet data found or invalid format received.")
+            return
+
+        # Prepare invoice configuration based on timesheet data
+        invoice_logo = os.getenv("INVOICE_LOGO", "/path/to/logo.png")
+        invoice_from = os.getenv("INVOICE_FROM", "Default Company")
+        invoice_to = os.getenv("INVOICE_TO", "Client Company")
+        invoice_tax = float(os.getenv("INVOICE_TAX", "0.0"))
+        invoice_rate = float(os.getenv("INVOICE_RATE", "25"))
+
+        config = {
+            "logo": invoice_logo,
+            "from": invoice_from,
+            "to": invoice_to,
+            "tax": invoice_tax,
+            "items": [],
+            "quantities": [],
+            "rates": [],
+        }
+
+        # For each timesheet entry, we'll use the note as item.
+        # If note is missing or "N/A", we'll use the date instead.
+        for entry in times:
+            entry_date = datetime.fromtimestamp(int(entry["date"])).strftime("%Y-%m-%d")
+            note = entry.get("note", "N/A")
+            item = note if note != "N/A" else f"Work on {entry_date}"
+            minutes = entry.get("minutes", 0)
+            config["items"].append(item)
+            config["quantities"].append(minutes)
+            config["rates"].append(invoice_rate)
+
+        # Generate YAML output
+        yaml_lines = [
+            f"logo: {config['logo']}",
+            f"from: {config['from']}",
+            f"to: {config['to']}",
+            f"tax: {config['tax']}",
+            "items:",
+        ]
+        for item in config["items"]:
+            yaml_lines.append(f'  - "{item}"')
+        yaml_lines.append("quantities:")
+        for qty in config["quantities"]:
+            yaml_lines.append(f"  - {qty}")
+        yaml_lines.append("rates:")
+        for rate in config["rates"]:
+            yaml_lines.append(f"  - {rate}")
+
+        yaml_content = "\n".join(yaml_lines)
+        config_filename = "invoice_config.yaml"
+
+        with open(config_filename, "w") as f:
+            f.write(yaml_content)
+
+        print(f"Invoice configuration written to {config_filename}")
+
+        # Call the invoice command if available locally
+        if shutil.which("invoice"):
+            os.system(
+                f"invoice generate --import {config_filename} --output timesheet-invoice.pdf"
+            )
+        else:
+            print(
+                "Invoice command not found locally. Please install it to generate invoices."
+            )
+
+    except Exception as e:
+        print(f"Error fetching timesheet data: {e}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())