Search code examples
sqlitepython-3.xtwitterpeewee

writing to and reading peewee


I am writing a program that scrapes tweets of a number of people, if the body of the tweet is unique it will get stored in the sqlite database for that person. I have two files, one to write to the databases and one to read the database for and search for tweets with a search word. Before writing to databases I printed the tweets on the terminal, the tweets are being pulled from twitter correctly. When I try a search a term all databases have zero tweets, even if there is no term. There is either a problem with the writing or reading of the database. Please help, I appreciate that I am very new to python.

the writing file:

import requests
import datetime
from bs4 import BeautifulSoup
from peewee import *
from time import sleep

databases = ["femfreq.db", "boris_johnson.db", "barack_obama.db",
         "daily_mail.db", "guardian.db", "times.db", "zac_goldsmith.db",
         "bernie_sanders.db", "george_osborne.db", "john_mcdonnell.db",
         "donald_trump.db", "hillary_clinton.db", "nigel_farage.db"]

urls = ["https://twitter.com/femfreq", "https://twitter.com/BorisJohnson",
    "https://twitter.com/BarackObama",
    "https://twitter.com/MailOnline?ref_src=twsrc%5Egoogle%7Ctwcamp%5Eserp%7Ctwgr%5Eauthor",
    "https://twitter.com/guardian?ref_src=twsrc%5Egoogle%7Ctwcamp%5Eserp%7Ctwgr%5Eauthor",
    "https://twitter.com/thetimes",
    "https://twitter.com/ZacGoldsmith?ref_src=twsrc%5Egoogle%7Ctwcamp%5Eserp%7Ctwgr%5Eauthor",
    "https://twitter.com/berniesanders?lang=en-gb",
    "https://twitter.com/George_Osborne?ref_src=twsrc%5Egoogle%7Ctwcamp%5Eserp%7Ctwgr%5Eauthor",
    "https://twitter.com/johnmcdonnellMP?ref_src=twsrc%5Egoogle%7Ctwcamp%5Eserp%7Ctwgr%5Eauthor",
    "https://twitter.com/realDonaldTrump?ref_src=twsrc%5Egoogle%7Ctwcamp%5Eserp%7Ctwgr%5Eauthor",
    "https://twitter.com/HillaryClinton?ref_src=twsrc%5Egoogle%7Ctwcamp%5Eserp%7Ctwgr%5Eauthor"
    "https://twitter.com/Nigel_Farage?ref_src=twsrc%5Egoogle%7Ctwcamp%5Eserp%7Ctwgr%5Eauthor"]

selection = 0

for database_chosen in databases:

    r = requests.get(urls[selection])
    soup = BeautifulSoup(r.content, "html.parser")
    content =soup.find_all("div",
              {"class":
               "content"})

    db = SqliteDatabase(database_chosen)


    class data_input(Model):
        time_position = DateTimeField(default=datetime.datetime.now)
        header = CharField()
        time_posted = CharField()
        tweet_body = CharField(unique=True)
        class Meta:
            database = db

    db.connect()
    db.create_tables([data_input], safe=True)


    for i in content:
        try:
            data_input.create(header = i.contents[1].text,
                          time_posted = i.contents[3].text,
                          tweet_body = i.contents[5].text)

        except IntegrityError:
            pass

    for i in content:
        print("=============")
        print(i.contents[1].text)
        print(i.contents[3].text)
        print(i.contents[5].text)


    selection += 1

    print("database: {} updated".format(database_chosen))

For the reading file

from peewee import *
import datetime

databases = ["femfreq.db", "boris_johnson.db", "barack_obama.db",
         "daily_mail.db", "guardian.db", "times.db", "zac_goldsmith.db",
         "bernie_sanders.db", "george_osborne.db", "john_mcdonnell.db",
         "donald_trump.db", "hillary_clinton.db", "nigel_farage.db"]

search_results = []


search_index = 0
print("")
print("Please enter the number for the database you want to search: ")
for i in databases:
    print("{}:{}".format(i, search_index))
    search_index += 1

select = int(input("please select: "))

database_chosen = databases[select]

db = SqliteDatabase(database_chosen)

class data_input(Model):
    time_position = DateTimeField(default=datetime.datetime.now)
    header = CharField()
    time_posted = CharField()
    tweet_body = CharField(unique=True)
    class Meta:
        database = db

db.connect()


enteries = data_input.select().order_by(data_input.time_position.desc())

print(enteries)

enteries = enteries.where(data_input.tweet_body)
print("")
print("The total number of tweets in {} is: {}".format(database_chosen,
                      len(enteries)))

For the reading file I haven't put in a search function yet I will move to that when I can get this problem first. Many thanks


Solution

  • What are you intending to accomplish by putting ".where(data_input.tweet_body)" in the query to read entries? Try removing that whole line:

    entries = entries.where(data_input.tweet_body)
    

    When you go to add your search, at that time you will want to add a where clause...something like:

    entries = entries.where(data_input.tweet_body.contains(search_term))