Add organization name field and additional target fields

- Add name column to Organization model
- Update database plugin to handle organization names
- Add migration for organization name field
- Add target fields: average_payout, start_date, end_date, is_updated
- Update upsert logic to include new fields

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>

Author: Derb237

src/synack/db/alembic/versions/8b478a84c1a6_add_organization_name.py

@@ -0,0 +1,26 @@
+"""add organization name
+
+Revision ID: 8b478a84c1a6
+Revises: 6f542023f57e
+Create Date: 2025-02-11 13:05:40.939271
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = '8b478a84c1a6'
+down_revision = '6f542023f57e'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    with op.batch_alter_table('organizations') as batch_op:
+        batch_op.add_column(sa.Column('name', sa.VARCHAR(100)))
+
+
+def downgrade():
+    with op.batch_alter_table('organizations') as batch_op:
+        batch_op.drop_column('name')

src/synack/db/models/organization.py

@@ -12,3 +12,4 @@
 class Organization(Base):
     __tablename__ = 'organizations'
     slug = sa.Column(sa.VARCHAR(20), primary_key=True)
+    name = sa.Column(sa.VARCHAR(100))

src/synack/plugins/db.py

@@ -104,15 +104,24 @@ def add_organizations(self, targets, session=None):
         for target in targets:
             if isinstance(target.get('organization'), str):
                 slug = target.get('organization')
+                name = None  # No name available in this case
             else:
-                slug = target.get('organization_id', target.get('organization', {}).get('slug'))
+                org = target.get('organization', {})
+                slug = target.get('organization_id', org.get('slug'))
+                name = org.get('name')
             if slug:
-                organizations_data.append({'slug': slug})
+                organizations_data.append({
+                    'slug': slug,
+                    'name': name
+                })
 
         if organizations_data:
             stmt = sqlite_insert(Organization).values(organizations_data)
-            stmt = stmt.on_conflict_do_nothing(
+            stmt = stmt.on_conflict_do_update(
                 index_elements=['slug'],
+                set_={
+                    'name': stmt.excluded.name
+                }
             )
             session.execute(stmt)
 
@@ -200,7 +209,11 @@ def add_targets(self, targets, **kwargs):
                     'is_active': target.get('isActive', target.get('is_active')),
                     'is_new': target.get('isNew', target.get('is_new')),
                     'is_registered': target.get('isRegistered', target.get('is_registered')),
-                    'last_submitted': target.get('lastSubmitted', target.get('last_submitted'))
+                    'last_submitted': target.get('lastSubmitted', target.get('last_submitted')),
+                    'average_payout': target.get('averagePayout'),
+                    'start_date': target.get('start_date'),
+                    'end_date': target.get('end_date'),
+                    'is_updated': target.get('isUpdated', False)
                 }
                 target_data.update(kwargs)
                 targets_data.append(target_data)
@@ -218,6 +231,10 @@ def add_targets(self, targets, **kwargs):
                     'is_new': stmt.excluded.is_new,
                     'is_registered': stmt.excluded.is_registered,
                     'last_submitted': stmt.excluded.last_submitted,
+                    'average_payout': stmt.excluded.average_payout,
+                    'start_date': stmt.excluded.start_date,
+                    'end_date': stmt.excluded.end_date,
+                    'is_updated': stmt.excluded.is_updated
                 }
             )
             session.execute(stmt)
@@ -378,7 +395,7 @@ def find_targets(self, **kwargs):
                 query = query.filter(sa.or_(*filters))
             else:
                 query = query.filter(sa.and_(*filters))
-    
+
         targets = query.all()
 
         session.expunge_all()