fixed number of characters killing

callmedeeray · Mar 11, 2019 · bbdc8a2 · bbdc8a2
1 parent 397bf43
commit bbdc8a2
Showing 1 changed file with 18 additions and 3 deletions.
diff --git a/scraper.py b/scraper.py
@@ -34,9 +34,24 @@
                 ind2 = k.find(')')
                 epinfo = k[ind+1:ind2]
                 killedwhat = k[ind2+2:len(k)]
-                dat = {'character': char, 'episode_info': episode, 'killed_what': killedwhat, 'kills': 1.0, 'details': k, 'pk': pk}
-                scraperwiki.sqlite.save(unique_keys=['pk'], data = dat)
-                pk += 1
+                if char.find(',') > -1:
+                    ch = char.split(',')
+                    for c in ch:
+                        char = c.replace('and','').strip()
+                        dat = {'character': c, 'episode_info': episode, 'killed_what': killedwhat, 'kills': 1.0/len(ch), 'details': k, 'pk': pk}
+                        scraperwiki.sqlite.save(unique_keys=['pk'], data = dat)
+                        pk += 1
+                else if char.find(',') == -1 and char.find(' and ') > -1:
+                    ch = char.split(' and ')
+                    for c in ch:
+                        char = c.replace(' and ','').strip()
+                        dat = {'character': c, 'episode_info': episode, 'killed_what': killedwhat, 'kills': 1.0/len(ch), 'details': k, 'pk': pk}
+                        scraperwiki.sqlite.save(unique_keys=['pk'], data = dat)
+                        pk += 1
+                else:
+                    dat = {'character': char, 'episode_info': episode, 'killed_what': killedwhat, 'kills': 1.0, 'details': k, 'pk': pk}
+                    scraperwiki.sqlite.save(unique_keys=['pk'], data = dat)
+                    pk += 1