From ba36cc7cbc538408765a0618c8d741e6baefe5d9 Mon Sep 17 00:00:00 2001
From: Jan Kaluza <jkaluza@redhat.com>
Date: Jan 16 2018 12:40:26 +0000
Subject: Fix traceback when upgrading db with events with NULL compose_id.


---

diff --git a/freshmaker/migrations/versions/b17231ee8220_remove_event_compose_id.py b/freshmaker/migrations/versions/b17231ee8220_remove_event_compose_id.py
index ce8bfa8..3318cca 100644
--- a/freshmaker/migrations/versions/b17231ee8220_remove_event_compose_id.py
+++ b/freshmaker/migrations/versions/b17231ee8220_remove_event_compose_id.py
@@ -32,6 +32,10 @@ def upgrade():
     for row in connection.execute('SELECT id, compose_id FROM events'):
         event_id, odcs_compose_id = row
 
+        # Skip the events with NULL/None compose_id.
+        if odcs_compose_id is None:
+            continue
+
         logger.info('Create Compose with odcs_compose_id %s from Event %s',
                     odcs_compose_id, event_id)
         connection.execute(
@@ -99,4 +103,4 @@ def downgrade():
         logger.info('Clear data from ArtifactBuildCompose')
         connection.execute('DELETE FROM artifact_build_composes')
         logger.info('Clear data from Compose')
-        connection.execute('DELETE FROM composes')
\ No newline at end of file
+        connection.execute('DELETE FROM composes')