Skip to content

Added fetch function as an out-of-band utility to easily fetch invoices #231

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
+49 −0
Open

Added fetch function as an out-of-band utility to easily fetch invoices #231

Changes from 1 commit
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
8636696
Closes #92
jimmysway Aug 19, 2025
1368d46
Added test for testing fetch out-of-band utility
jimmysway Sep 5, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 18 additions & 0 deletions process_report/invoices/invoice.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
from dataclasses import dataclass
import pandas
import os

import process_report.util as util

Expand Down Expand Up @@ -122,3 +123,20 @@ def export(self):
def export_s3(self, s3_bucket):
s3_bucket.upload_file(self.output_path, self.output_s3_key)
s3_bucket.upload_file(self.output_path, self.output_s3_archive_key)

def fetch(self, s3_bucket=None):
"""Fetches the exported invoice CSV from S3 and loads it into self.data.

If s3_bucket is not provided, uses the default invoice bucket.
The CSV is downloaded to the current working directory using its base filename.
"""
s3_key = self.output_s3_key
local_filename = os.path.basename(s3_key)

if s3_bucket is None:
# Fallback to default bucket utility which also downloads
util.fetch_s3(s3_key)
else:
s3_bucket.download_file(s3_key, local_filename)

self.data = pandas.read_csv(local_filename)
Loading