For startup crashes, use the StartupCrash annotation when available (>= 51)

generate-data.py

@@ -113,6 +113,12 @@ def get(channel, date, product='Firefox', duration=11, tc_limit=50, crash_type='
                 elif platform['term'] == 'Mac OS X':
                     signatures[signature['term']][2] = platform['count']
 
+            # XXX: Remove this when all versions will have the StartupCrash annotation.
+            if version >= 51:
+                for startup_crash in signature['facets']['startup_crash']:
+                    if startup_crash in ['1', 'T']:
+                        signatures[signature['term']][4] += startup_crash['count']
+            else:
                 for uptime in signature['facets']['uptime']:
                     if int(uptime['term']) < 60:
                         signatures[signature['term']][4] += uptime['count']
@@ -127,7 +133,7 @@ def get(channel, date, product='Firefox', duration=11, tc_limit=50, crash_type='
         'version': versions,
         'date': socorro.SuperSearch.get_search_date(start_date, end_date),
         'release_channel': channel,
-        '_aggs.signature': ['platform', 'uptime', '_cardinality.install_time'],
+        '_aggs.signature': ['platform', 'uptime', '_cardinality.install_time', 'startup_crash'],
         '_results_number': 0,
         '_facets_size': tc_limit,
         '_histogram.date': ['product'],
@@ -135,6 +141,10 @@ def get(channel, date, product='Firefox', duration=11, tc_limit=50, crash_type='
     }
 
     if startup:
+        # XXX: Remove this when all versions will have the StartupCrash annotation.
+        if version >= 51:
+            params['startup_crash'] = True
+        else:
             params['uptime'] = '<=60'
 
     sys.stdout.write('Getting top signatures from Socorro...')