Add mentions field

387245a6 · Cody Zacharias · GitHub · 1cb40f0c · 387245a6
Commit 387245a6 authored May 27, 2018 by Cody Zacharias Committed by GitHub May 27, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 3 deletions

twint/db.py twint/db.py +5 -3

No files found.
--- a/twint/db.py
+++ b/twint/db.py
@@ -22,7 +22,8 @@ def init(db):
                    hashtags text,
                    link text
                    retweet bool,
-                    user_rt text
+                    user_rt text,
+                    mentions text
                );
        """
        cursor.execute(table_tweets)
@@ -111,8 +112,9 @@ def tweets(conn, Tweet):
                    ",".join(Tweet.hashtags),
                    Tweet.link,
                    Tweet.is_retweet,
-                    Tweet.user_rt)
-        cursor.execute('INSERT INTO tweets VALUES(?,?,?,?,?,?,?,?,?,?,?,?,?)', entry)
+                    Tweet.user_rt,
+                    ",".join(Tweet.mentions))
+        cursor.execute('INSERT INTO tweets VALUES(?,?,?,?,?,?,?,?,?,?,?,?,?,?)', entry)
        conn.commit()
    except sqlite3.IntegrityError:
        pass