twitools/tools.py

import config

import csv, datetime, os, sqlite3, sys, tweepy

class dbObject:

 def __init__(self, path=config.dbpath):
  self.conn = sqlite3.connect(path)
  self.cur = self.conn.cursor() 
  self.path = path

 def closeConnection(self):
  return self.conn.close()

 def commit(self):
  return self.conn.commit()

 def executeQuery(self, query):
  return self.cur.execute(query)

 def getNext(self):
  return self.cur.fetchone()

 def isInitialized(self):
  try:
   self.executeQuery("SELECT * FROM tweets")
   return True
  except:
   return False

 def getFLDate(self, val = 0):
  if val == 0:
   mode = "MIN"
  else:
   mode = "MAX"

  return getDate(str(list(self.executeQuery("SELECT %s(SUBSTR(timestamp,0,11)) FROM tweets" % mode))[0][0]))


class twObject:

 def __init__(self, cke = config.cke, cse = config.cse, ato = config.ato, ase = config.ase):
  self.auth = tweepy.OAuthHandler(config.cke, config.cse)
  self.auth.set_access_token(config.ato, config.ase)
  self.api = tweepy.API(self.auth)

 def search(self, query, savepoint = 0):
  tweets = list(tweepy.Cursor(self.api.search, q=query, since_id=savepoint).items())
  tweets.reverse()
  return tweets


def dbCheck(db, create = False):
 if (not create and dbInitialized(db)) or (create and not dbInitialized(db)):
  return True
 if create:
  raise ValueError("Provided database file " + db.path + " is already initialized. Remove it manually before trying to recreate it.")
 raise ValueError("Provided database file " + db.path + " is not initialized. Create it using makedb.py or csvdb.py")


def dbHelper(path, create = False):
 db = dbObject(path)
 dbCheck(db, create)
 return db


def dbInitialized(db):
 return db.isInitialized()


def fileExists(path):
 return os.path.isfile(path)


def getDate(date):
 try:
  return datetime.datetime.strptime(date, '%Y-%m-%d')
 except ValueError:
  raise ValueError("Dates must be in YYYY-MM-DD format.")


def parseArgs(argv):
 args = []
 path = None
 nextpath = False

 for a in argv[1:]:
  if nextpath:
   path = a
   nextpath = False
  elif a == "-f":
   if path != None:
    raise ValueError("You can only pass one database file.")
   nextpath = True
  else:
   args += [a]

 return args, path

def printCSV(inlist):
 writer = csv.writer(sys.stdout)
 writer.writerows(inlist)