twitools/gettweets.py

#!/usr/bin/env python3

import dbtools, setuptools

import sys, datetime

def dateArgs(argv = sys.argv[1:]):
 fr = None
 to = None

 mode = 0

 for arg in argv:
  if mode == 0:
   if arg == "-f":
    mode = 1
   elif arg == "-t":
    mode = 2
   else:
    strings += [arg]
    mode = 0
  elif mode == 1:
    fr = setuptools.getDate(arg)
    mode = 0
  else:
    to = setuptools.getDate(arg)
    mode = 0

 if mode in (1, 2):
  raise ValueError("Date missing.")

 if to != None and fr != None and to < fr:
  raise ValueError("From date must be before To date.")

 return fr, to
    

def queryBuilder(fr, to):
 return "SELECT * FROM tweets WHERE SUBSTR(timestamp,0,11) >= '%s' AND SUBSTR(timestamp,0,11) <= '%s'" % (fr, to)


def getDataByDate(fr, to, db = dbtools.dbObject(), headers = True):
 if fr == None:
  fr = db.getFLDate()
 if to == None:
  to = db.getFLDate(1)

 cur = list(db.executeQuery(queryBuilder(fr, to)))

 if headers:
  pass

 return cur


if __name__ == "__main__":
 setuptools.printCSV(getDataByDate(*dateArgs(), headers = True))