mirror of
https://github.com/NNTmux/newznab-tmux.git
synced 2026-08-30 01:38:54 +00:00
Move files to logical positions, no more copying of files.
This commit is contained in:
@@ -0,0 +1,72 @@
|
||||
#!/usr/bin/python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import string
|
||||
import os, sys, re
|
||||
|
||||
class bcolors:
|
||||
HEADER = '\033[38;5;011m'
|
||||
PRIMARY = '\033[38;5;010m'
|
||||
ERROR = '\033[31mError: '
|
||||
ENDC = '\033[0m'
|
||||
ALTERNATE = '\033[38;5;199m'
|
||||
def disable(self):
|
||||
self.HEADER = ''
|
||||
self.PRIMARY = ''
|
||||
|
||||
pathname = os.path.abspath(os.path.dirname(sys.argv[0]))
|
||||
def connect():
|
||||
conf = readConfig()
|
||||
con = None
|
||||
if conf['DB_TYPE'] == "mysql":
|
||||
try:
|
||||
import cymysql as mdb
|
||||
if conf['DB_PORT'] != '':
|
||||
con = mdb.connect(host=conf['DB_HOST'], user=conf['DB_USER'], passwd=conf['DB_PASSWORD'], db=conf['DB_NAME'], port=int(conf['DB_PORT']), charset="utf8")
|
||||
else:
|
||||
con = mdb.connect(host=conf['DB_HOST'], user=conf['DB_USER'], passwd=conf['DB_PASSWORD'], db=conf['DB_NAME'], charset="utf8")
|
||||
except ImportError:
|
||||
print(bcolors.ERROR + "\nPlease install cymysql for python 3, \ninformation can be found in threaded_scripts_readme.txt\n" + bcolors.ENDC)
|
||||
sys.exit()
|
||||
elif conf['DB_TYPE'] == "pgsql":
|
||||
try:
|
||||
import psycopg2 as mdb
|
||||
con = mdb.connect(host=conf['DB_HOST'], user=conf['DB_USER'], password=conf['DB_PASSWORD'], dbname=conf['DB_NAME'], port=int(conf['DB_PORT']))
|
||||
except ImportError:
|
||||
print(bcolors.HEADER + "\nPlease install psycopg for python 3, \ninformation can be found in threaded_scripts_readme.txt\n" + bcolors.ENDC)
|
||||
sys.exit()
|
||||
cur = con.cursor()
|
||||
return cur, con
|
||||
|
||||
def disconnect(cur, con):
|
||||
con.close()
|
||||
con = None
|
||||
cur.close()
|
||||
cur = None
|
||||
|
||||
def readConfig():
|
||||
Configfile = pathname+"/../../../../../www/config.php"
|
||||
file = open( Configfile, "r")
|
||||
|
||||
# Match a config line
|
||||
m = re.compile('^define\(\'([A-Z_]+)\', \'?(.*?)\'?\);$', re.I)
|
||||
|
||||
# The config object
|
||||
config = {}
|
||||
for line in file.readlines():
|
||||
match = m.search( line )
|
||||
if match:
|
||||
value = match.group(2)
|
||||
|
||||
# filter boolean
|
||||
if "true" is value:
|
||||
value = True
|
||||
elif "false" is value:
|
||||
value = False
|
||||
|
||||
# Add to the config
|
||||
config[ match.group(1) ] = value
|
||||
return config
|
||||
|
||||
# Test
|
||||
config = readConfig()
|
||||
Reference in New Issue
Block a user