twitools/setuptools/__init__.py

import ast, configparser, csv, datetime, html.parser, itertools, os, sqlite3, sys, tweepy

TWITTER = 0
MARKOV = 1
LYRICS = 2

class SetupException(Exception):
 def __str__(self):
  return "Seems like config.cfg has not been created yet or contains serious errors. Run setup.py to create it."

def getSetting(section, setting, path = "config.cfg"):
 config = configparser.RawConfigParser()
 config.read(path)
 return config.get(section, setting)

def getListSetting(section, setting, path = "config.cfg"):
 return ast.literal_eval(getSetting(section, setting, path))

def dbtype():
 try:
  return int(getSetting("Database", "type"))
 except:
  return 0 # for SQLite3

### Must only be called AFTER dbtype()! ###

def dbhost():
 try:
  return getSetting("Database", "host")
 except:
  raise SetupException()

def dbuser():
 try:
  return getSetting("Database", "user")
 except:
  raise SetupException()

def dbpass():
 try:
  return getSetting("Database", "pass")
 except:
  raise SetupException()

def dbname():
 try:
  return getSetting("Database", "name")
 except:
  raise SetupException()

def dbpath():
 try:
  return getSetting("Database", "path")
 except:
  return SetupException()

###

def sectionName(section):
 if section == TWITTER:
  return "Twitter"
 elif section == MARKOV:
  return "Markov"
 elif section == LYRICS:
  return "Lyrics"

def cke():
 try:
  return getSetting("Twitter", "cke")
 except:
  raise SetupException()

def cse():
 try:
  return getSetting("Twitter", "cse")
 except:
  raise SetupException()

def ato(section = TWITTER):
 try:
  return getSetting(sectionName(section), "ato")
 except:
  raise SetupException()

def ase(section = TWITTER):
 try:
  return getSetting(sectionName(section), "ase")
 except:
  raise SetupException()

def botname():
 try: 
  return getSetting("Telegram", "botname")
 except:
  return SetupException()

def token():
 try:
  return getSetting("Telegram", "token")
 except:
  return SetupException()

def url():
 try:
  return getSetting("Bot", "url")
 except:
  return SetupExecption()

def admin():
 try:
  return getSetting("Bot", "admin")
 except:
  return SetupException()

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 paginate(iterable, page_size):
 while True:
  i1, i2 = itertools.tee(iterable)
  iterable, page = (itertools.islice(i1, page_size, None), list(itertools.islice(i2, page_size)))
  if len(page) == 0:
   break
  yield page


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)

def unescapeText(text):
 return html.parser.HTMLParser().unescape(text).replace("'","''")