Skip to content

Commit

Permalink
updating naming
Browse files Browse the repository at this point in the history
  • Loading branch information
kylevillegas93 committed Nov 25, 2024
1 parent caaf078 commit 5215423
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions processes/record_buffer.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,17 +9,17 @@ def __init__(self, db_manager: DBManager, batch_size: int=500):
self.batch_size = batch_size
self.ingest_count = 0

def add(self, record: BaseMapping):
def add(self, record_mapping: BaseMapping):
existing_record = self.db_manager.session.query(Record).filter(
Record.source_id == record.record.source_id
Record.source_id == record_mapping.record.source_id
).first()

if existing_record:
record.updateExisting(existing_record)
record_mapping.updateExisting(existing_record)
self.records.discard(existing_record)
self.records.add(existing_record)
else:
self.records.add(record.record)
self.records.add(record_mapping.record)

if self.batch_size >= len(self.records):
self.flush()
Expand Down

0 comments on commit 5215423

Please sign in to comment.