diff --git a/lyricsbot.py b/lyricsbot.py index 6da83b1..5a0e762 100755 --- a/lyricsbot.py +++ b/lyricsbot.py @@ -10,10 +10,18 @@ def getLyrics(db = dbtools.dbHelper()): return lyric[0], lyric[1], ref +def findParent(lid, db = dbtools.dbHelper()): + ref = int(list(db.executeQuery("SELECT ref FROM lyrics WHERE id = %i;" % int(lid)))[0][0]) + return lid if ref == 0 else findParent(ref) + def postprocess(lid, tid, db = dbtools.dbHelper()): db.executeQuery("UPDATE lyrics SET tweet_id = %i WHERE id = %i;" % (int(tid), int(lid))) - db.executeQuery("UPDATE lyrics SET active = 1 WHERE ref = %i;" % int(lid)) db.executeQuery("UPDATE lyrics SET active = 0 WHERE id = %i;" % int(lid)) + + if list(db.executeQuery("SELECT COUNT(*) FROM lyrics WHERE ref = %i;" % int(lid)))[0][0] == 0: + db.executeQuery("UPDATE lyrics SET active = 1 WHERE id = %i;" % findParent(lid)) + else: + db.executeQuery("UPDATE lyrics SET active = 1 WHERE ref = %i;" % int(lid)) db.commit() def tweet(text, ref = 0, two = twitools.twObject()):