60 lines
1 KiB
Python
60 lines
1 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import tools
|
||
|
|
|
||
|
|
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 = tools.getDate(arg)
|
||
|
|
mode = 0
|
||
|
|
else:
|
||
|
|
to = tools.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, path = tools.config.dbpath, headers = True):
|
||
|
|
db = tools.dbHelper(path)
|
||
|
|
|
||
|
|
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__":
|
||
|
|
tools.printCSV(getDataByDate(*dateArgs(), headers = True))
|