mirror of
https://github.com/NNTmux/newznab-tmux.git
synced 2026-08-29 17:28:55 +00:00
Move files to logical positions, no more copying of files.
This commit is contained in:
@@ -0,0 +1,191 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import sys, os, time
|
||||
import threading
|
||||
try:
|
||||
import queue
|
||||
except ImportError:
|
||||
import Queue as queue
|
||||
import subprocess
|
||||
import string
|
||||
import signal
|
||||
import datetime
|
||||
import math
|
||||
|
||||
import lib.info as info
|
||||
from lib.info import bcolors
|
||||
conf = info.readConfig()
|
||||
cur = info.connect()
|
||||
start_time = time.time()
|
||||
pathname = os.path.abspath(os.path.dirname(sys.argv[0]))
|
||||
|
||||
print(bcolors.HEADER + "\nBackfill Safe Threaded Started at {}".format(datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
|
||||
cur[0].execute("SELECT g.name FROM groups g LEFT JOIN shortgroups ON shortgroups.name = g.name WHERE shortgroups.name IS NULL AND backfill = 1")
|
||||
dorun = cur[0].fetchone()
|
||||
|
||||
#close connection to mysql
|
||||
info.disconnect(cur[0], cur[1])
|
||||
|
||||
if dorun:
|
||||
#before we get the groups, lets update shortgroups
|
||||
subprocess.call(["php", pathname+"/../bin/update_groups.php", ""])
|
||||
else:
|
||||
cur = info.connect()
|
||||
cur[0].execute("SELECT name FROM shortgroups")
|
||||
dorun = cur[0].fetchone()
|
||||
info.disconnect(cur[0], cur[1])
|
||||
if len(sys.argv) > 1 and sys.argv[1] not in dorun:
|
||||
#before we get the groups, lets update shortgroups
|
||||
subprocess.call(["php", pathname+"/../bin/update_groups.php", ""])
|
||||
|
||||
count = 0
|
||||
previous = "'alt.binaries.crap'"
|
||||
|
||||
#if the group has less than 10000 to grab, just grab them, and loop another group
|
||||
while count < 10000:
|
||||
#get values from db
|
||||
cur = info.connect()
|
||||
cur[0].execute("SELECT (SELECT value FROM site WHERE setting = 'backfillthreads') AS a, (SELECT value FROM tmux WHERE setting = 'backfill_qty') AS b, (SELECT value FROM tmux WHERE setting = 'backfill') AS c, (SELECT value FROM tmux WHERE setting = 'backfill_order') AS e, (SELECT value FROM tmux WHERE setting = 'backfill_days') AS f, (SELECT value FROM site WHERE setting = 'maxmssgs') AS g")
|
||||
dbgrab = cur[0].fetchall()
|
||||
run_threads = int(dbgrab[0][0])
|
||||
backfill_qty = int(dbgrab[0][1])
|
||||
type = int(dbgrab[0][2])
|
||||
intorder = int(dbgrab[0][3])
|
||||
intbackfilltype = int(dbgrab[0][4])
|
||||
maxmssgs = int(dbgrab[0][5])
|
||||
|
||||
#get the correct oder by for the query
|
||||
if intorder == 1:
|
||||
group = "ORDER BY first_record_postdate DESC"
|
||||
elif intorder == 2:
|
||||
group = "ORDER BY first_record_postdate ASC"
|
||||
elif intorder == 3:
|
||||
group = "ORDER BY name ASC"
|
||||
elif intorder == 4:
|
||||
group = "ORDER BY name DESC"
|
||||
elif intorder == 5:
|
||||
group = "ORDER BY a.last_record DESC"
|
||||
else:
|
||||
group = "ORDER BY a.last_record ASC"
|
||||
|
||||
#backfill days or safe backfill date
|
||||
if intbackfilltype == 1:
|
||||
backfilldays = "backfill_target"
|
||||
elif intbackfilltype == 2:
|
||||
backfilldays = "datediff(curdate(),(SELECT value FROM site WHERE setting = 'safebackfilldate'))"
|
||||
|
||||
#query to grab backfill groups
|
||||
if len(sys.argv) == 1:
|
||||
if conf['DB_TYPE'] == "mysql":
|
||||
cur[0].execute("SELECT g.name, g.first_record AS our_first, MAX(a.first_record) AS thier_first, MAX(a.last_record) AS their_last FROM groups g INNER JOIN shortgroups a ON g.name = a.name WHERE g.first_record IS NOT NULL AND g.first_record_postdate IS NOT NULL AND g.backfill = 1 AND (NOW() - INTERVAL %s DAY) < g.first_record_postdate AND g.name NOT IN (%s) GROUP BY a.name, a.last_record, g.name, g.first_record %s LIMIT 1" % (backfilldays, previous, group))
|
||||
elif conf['DB_TYPE'] == "pgsql":
|
||||
cur[0].execute("SELECT g.name, g.first_record AS our_first, MAX(a.first_record) AS thier_first, MAX(a.last_record) AS their_last FROM groups g INNER JOIN shortgroups a ON g.name = a.name WHERE g.first_record IS NOT NULL AND g.first_record_postdate IS NOT NULL AND g.backfill = 1 AND (NOW() - INTERVAL '%s DAYS') < g.first_record_postdate GROUP BY a.name, a.last_record, g.name, g.first_record %s LIMIT 1" % (backfilldays, group, groups))
|
||||
datas = cur[0].fetchone()
|
||||
else:
|
||||
run = "SELECT g.name, g.first_record AS our_first, MAX(a.first_record) AS thier_first, MAX(a.last_record) AS their_last FROM groups g INNER JOIN shortgroups a ON g.name = a.name WHERE g.name = %s AND g.first_record IS NOT NULL AND g.first_record_postdate IS NOT NULL AND g.backfill = 1 LIMIT 1"
|
||||
cur[0].execute(run, (sys.argv[1]))
|
||||
datas = cur[0].fetchone()
|
||||
if not datas or datas[0] is None:
|
||||
print(bcolors.ERROR + "No Groups enabled for backfill" + bcolors.ENDC)
|
||||
info.disconnect(cur[0], cur[1])
|
||||
sys.exit()
|
||||
|
||||
#close connection to mysql
|
||||
info.disconnect(cur[0], cur[1])
|
||||
|
||||
previous += ", '%s'" % datas[0]
|
||||
count = datas[1] - datas[2]
|
||||
if count < 0:
|
||||
print(bcolors.ERROR + "USP returned an invalid first_post for {}, skipping it.".format(datas[0]) + bcolors.ENDC)
|
||||
if len(sys.argv) == 2:
|
||||
sys.exit()
|
||||
|
||||
if count == 0:
|
||||
if len(sys.argv) == 2:
|
||||
print(bcolors.ERROR + "We have hit the maximum we can backfill for {}, disabling it".format(datas[0]) + bcolors.ENDC)
|
||||
remove = "UPDATE groups SET backfill = 0 WHERE name = %s"
|
||||
cur = info.connect()
|
||||
cur[0].execute(remove, (sys.argv[1]))
|
||||
cur[1].autocommit(True)
|
||||
info.disconnect(cur[0], cur[1])
|
||||
sys.exit()
|
||||
else:
|
||||
print(bcolors.ERROR + "We have hit the maximum we can backfill for {}, skipping it".format(datas[0]) + bcolors.ENDC)
|
||||
|
||||
if count < 10000 and count > 0:
|
||||
print(bcolors.PRIMARY + "Group {} has {} articles, in the range {} to {}".format(datas[0], "{:,}".format(count), "{:,}".format(datas[2]), "{:,}".format(datas[3])) + bcolors.ENDC)
|
||||
print(bcolors.PRIMARY + "Our oldest post is: {}".format("{:,}".format(datas[1])) + bcolors.ENDC)
|
||||
print(bcolors.PRIMARY + "Available Posts: {}".format("{:,}".format(count)) + bcolors.ENDC)
|
||||
group = ("{} {}".format(datas[0], count))
|
||||
subprocess.call(["php", pathname+"/../../multiprocessing/.do_not_run/switch.php", "python backfill_all_quantity "+str(group)])
|
||||
|
||||
#calculate the number of items for queue
|
||||
if (count > (backfill_qty * run_threads)):
|
||||
geteach = math.ceil((backfill_qty * run_threads) / maxmssgs)
|
||||
else:
|
||||
geteach = int(count / maxmssgs)
|
||||
|
||||
my_queue = queue.Queue()
|
||||
time_of_last_run = time.time()
|
||||
|
||||
class queue_runner(threading.Thread):
|
||||
def __init__(self, my_queue):
|
||||
threading.Thread.__init__(self)
|
||||
self.my_queue = my_queue
|
||||
|
||||
def run(self):
|
||||
global time_of_last_run
|
||||
|
||||
while True:
|
||||
try:
|
||||
my_id = self.my_queue.get(True, 1)
|
||||
except:
|
||||
if time.time() - time_of_last_run > 3:
|
||||
return
|
||||
else:
|
||||
if my_id:
|
||||
time_of_last_run = time.time()
|
||||
subprocess.call(["php", pathname+"/../../multiprocessing/.do_not_run/switch.php", "python "+my_id])
|
||||
time.sleep(.03)
|
||||
self.my_queue.task_done()
|
||||
|
||||
def main(args):
|
||||
global time_of_last_run
|
||||
time_of_last_run = time.time()
|
||||
|
||||
print(bcolors.HEADER + "We will be using a max of {} threads, a queue of {} and grabbing {} headers".format(run_threads, "{:,}".format(geteach), "{:,}".format(geteach * maxmssgs)) + bcolors.ENDC)
|
||||
time.sleep(2)
|
||||
|
||||
def signal_handler(signal, frame):
|
||||
sys.exit(0)
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
|
||||
if True:
|
||||
#spawn a pool of place worker threads
|
||||
for i in range(run_threads):
|
||||
p = queue_runner(my_queue)
|
||||
p.setDaemon(False)
|
||||
p.start()
|
||||
|
||||
#now load some arbitrary jobs into the queue
|
||||
for i in range(0, int(geteach)):
|
||||
time.sleep(.03)
|
||||
my_queue.put("get_range backfill %s %s %s %s" % (datas[0], datas[1] - i * maxmssgs - maxmssgs, datas[1] - i * maxmssgs - 1, i+1))
|
||||
|
||||
my_queue.join()
|
||||
|
||||
group = ("{} {}".format(datas[0], 1000))
|
||||
subprocess.call(["php", pathname+"/../../multiprocessing/.do_not_run/switch.php", "python backfill_all_quantity "+str(group)])
|
||||
if run_threads <= geteach:
|
||||
print(bcolors.HEADER + "\nWe used {} threads, a queue of {} and grabbed {} headers".format(run_threads, "{:,}".format(geteach), "{:,}".format(geteach * maxmssgs)) + bcolors.ENDC)
|
||||
else:
|
||||
print(bcolors.HEADER + "\nWe used {} threads, a queue of {} and grabbed {} headers".format(geteach, "{:,}".format(geteach), "{:,}".format(geteach * maxmssgs)) + bcolors.ENDC)
|
||||
|
||||
print(bcolors.HEADER + "\nBackfill Safe Threaded Completed at {}".format(datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
print(bcolors.HEADER + "Running time: {}\n\n".format(str(datetime.timedelta(seconds=time.time() - start_time))) + bcolors.ENDC)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main(sys.argv[1:])
|
||||
@@ -0,0 +1,138 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
from __future__ import print_function
|
||||
import sys, os, time
|
||||
import threading
|
||||
try:
|
||||
import queue
|
||||
except ImportError:
|
||||
import Queue as queue
|
||||
import subprocess
|
||||
import string
|
||||
import signal
|
||||
import datetime
|
||||
|
||||
import lib.info as info
|
||||
from lib.info import bcolors
|
||||
conf = info.readConfig()
|
||||
cur = info.connect()
|
||||
start_time = time.time()
|
||||
pathname = os.path.abspath(os.path.dirname(sys.argv[0]))
|
||||
|
||||
print(bcolors.HEADER + "\nBackfill Threaded Started at {}".format(datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
|
||||
#get values from db
|
||||
cur[0].execute("SELECT (SELECT value FROM site WHERE setting = 'backfillthreads') as a, (SELECT value FROM tmux WHERE setting = 'backfill') as c, (SELECT value FROM tmux WHERE setting = 'backfill_groups') as d, (SELECT value FROM tmux WHERE setting = 'backfill_order') as e, (SELECT value FROM tmux WHERE setting = 'backfill_days') as f")
|
||||
dbgrab = cur[0].fetchall()
|
||||
run_threads = int(dbgrab[0][0])
|
||||
type = int(dbgrab[0][1])
|
||||
groups = int(dbgrab[0][2])
|
||||
intorder = int(dbgrab[0][3])
|
||||
intbackfilltype = int(dbgrab[0][4])
|
||||
|
||||
#get the correct oder by for the query
|
||||
if intorder == 1:
|
||||
group = "ORDER BY first_record_postdate DESC"
|
||||
elif intorder == 2:
|
||||
group = "ORDER BY first_record_postdate ASC"
|
||||
elif intorder == 3:
|
||||
group = "ORDER BY name ASC"
|
||||
elif intorder == 4:
|
||||
group = "ORDER BY name DESC"
|
||||
elif intorder == 5:
|
||||
group = "ORDER BY first_record DESC"
|
||||
else:
|
||||
group = "ORDER BY first_record ASC"
|
||||
|
||||
#backfill days or safe backfill date
|
||||
if intbackfilltype == 1:
|
||||
backfilldays = "backfill_target"
|
||||
elif intbackfilltype == 2:
|
||||
backfilldays = "datediff(curdate(),(SELECT value FROM site WHERE setting = 'safebackfilldate'))"
|
||||
|
||||
#exit is set to safe backfill
|
||||
if len(sys.argv) == 1 and type == 4:
|
||||
print(bcolors.ERROR + "Tmux is set for Safe Backfill, no groups to process." + bcolors.ENDC)
|
||||
info.disconnect(cur[0], cur[1])
|
||||
sys.exit()
|
||||
|
||||
#query to grab backfill groups
|
||||
if len(sys.argv) > 1 and sys.argv[1] == "all":
|
||||
# Using string formatting is not the correct way to do this, but using +group is even worse
|
||||
# removing the % before the variables at the end of the query adds quotes/escapes strings
|
||||
cur[0].execute("SELECT name, first_record FROM groups WHERE first_record != 0 AND backfill = 1 %s" % (group))
|
||||
else:
|
||||
if conf['DB_TYPE'] == "mysql":
|
||||
cur[0].execute("SELECT name, first_record FROM groups WHERE first_record != 0 AND first_record_postdate IS NOT NULL AND backfill = 1 AND (NOW() - interval %s DAY) < first_record_postdate %s LIMIT %s" % (backfilldays, group, groups))
|
||||
elif conf['DB_TYPE'] == "pgsql":
|
||||
cur[0].execute("SELECT name, first_record FROM groups WHERE first_record != 0 AND first_record_postdate IS NOT NULL AND backfill = 1 AND (NOW() - interval '%s DAYS') < first_record_postdate %s LIMIT %s" % (backfilldays, group, groups))
|
||||
|
||||
datas = cur[0].fetchall()
|
||||
|
||||
#close connection to mysql
|
||||
info.disconnect(cur[0], cur[1])
|
||||
|
||||
if not datas:
|
||||
print(bcolors.ERROR + "No Groups enabled for backfill" + bcolors.ENDC)
|
||||
sys.exit()
|
||||
|
||||
|
||||
my_queue = queue.Queue()
|
||||
time_of_last_run = time.time()
|
||||
|
||||
class queue_runner(threading.Thread):
|
||||
def __init__(self, my_queue):
|
||||
threading.Thread.__init__(self)
|
||||
self.my_queue = my_queue
|
||||
|
||||
def run(self):
|
||||
global time_of_last_run
|
||||
|
||||
while True:
|
||||
try:
|
||||
my_id = self.my_queue.get(True, 1)
|
||||
except:
|
||||
if time.time() - time_of_last_run > 3:
|
||||
return
|
||||
else:
|
||||
if my_id:
|
||||
time_of_last_run = time.time()
|
||||
if len(sys.argv) > 1 and sys.argv[1] == "all":
|
||||
subprocess.call(["php", pathname+"/../../multiprocessing/.do_not_run/switch.php", "python backfill_all_quick "+my_id])
|
||||
else:
|
||||
subprocess.call(["php", pathname+"/../../multiprocessing/.do_not_run/switch.php", "python backfill "+my_id])
|
||||
time.sleep(.03)
|
||||
self.my_queue.task_done()
|
||||
|
||||
def main(args):
|
||||
global time_of_last_run
|
||||
time_of_last_run = time.time()
|
||||
|
||||
print(bcolors.HEADER + "We will be using a max of {} threads, a queue of {} groups".format(run_threads, "{:,}".format(len(datas))) + bcolors.ENDC)
|
||||
time.sleep(2)
|
||||
|
||||
def signal_handler(signal, frame):
|
||||
sys.exit()
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
|
||||
if True:
|
||||
#spawn a pool of place worker threads
|
||||
for i in range(run_threads):
|
||||
p = queue_runner(my_queue)
|
||||
p.setDaemon(False)
|
||||
p.start()
|
||||
|
||||
#now load some arbitrary jobs into the queue
|
||||
for gnames in datas:
|
||||
time.sleep(.03)
|
||||
my_queue.put("%s %s" % (gnames[0], type))
|
||||
|
||||
my_queue.join()
|
||||
|
||||
print(bcolors.HEADER + "\nBackfill Threaded Completed at {}".format(datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
print(bcolors.HEADER + "Running time: {}\n\n".format(str(datetime.timedelta(seconds=time.time() - start_time))) + bcolors.ENDC)
|
||||
|
||||
if __name__ == '__main__':
|
||||
main(sys.argv[1:])
|
||||
@@ -0,0 +1,124 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import sys, os, time
|
||||
import threading
|
||||
try:
|
||||
import queue
|
||||
except ImportError:
|
||||
import Queue as queue
|
||||
import subprocess
|
||||
import string
|
||||
import signal
|
||||
import datetime
|
||||
import math
|
||||
|
||||
import lib.info as info
|
||||
from lib.info import bcolors
|
||||
conf = info.readConfig()
|
||||
cur = info.connect()
|
||||
start_time = time.time()
|
||||
pathname = os.path.abspath(os.path.dirname(sys.argv[0]))
|
||||
count = 0
|
||||
|
||||
print(bcolors.HEADER + "\nBinaries Safe Threaded Started at {}".format(datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
|
||||
#get values from db
|
||||
cur[0].execute("SELECT (SELECT value FROM site WHERE setting = 'binarythreads') AS a, (SELECT value FROM site WHERE setting = 'maxmssgs') AS b")
|
||||
dbgrab = cur[0].fetchall()
|
||||
|
||||
run_threads = int(dbgrab[0][0])
|
||||
maxmssgs = int(dbgrab[0][1])
|
||||
|
||||
#before we get the groups, lets update shortgroups
|
||||
subprocess.call(["php", pathname+"/../bin/update_groups.php", ""])
|
||||
|
||||
#query to grab all active groups
|
||||
cur = info.connect()
|
||||
cur[0].execute("SELECT g.name AS groupname, g.last_record AS our_last, a.last_record AS their_last FROM groups g INNER JOIN shortgroups a ON g.active = 1 AND g.name = a.name ORDER BY a.last_record DESC")
|
||||
datas = cur[0].fetchall()
|
||||
|
||||
#close connection to mysql
|
||||
info.disconnect(cur[0], cur[1])
|
||||
|
||||
if not datas:
|
||||
print(bcolors.ERROR + "No Groups activated" + bcolors.ENDC)
|
||||
sys.exit()
|
||||
|
||||
my_queue = queue.Queue()
|
||||
time_of_last_run = time.time()
|
||||
|
||||
class queue_runner(threading.Thread):
|
||||
def __init__(self, my_queue):
|
||||
threading.Thread.__init__(self)
|
||||
self.my_queue = my_queue
|
||||
|
||||
def run(self):
|
||||
global time_of_last_run
|
||||
while True:
|
||||
try:
|
||||
my_id = self.my_queue.get(True, 1)
|
||||
except:
|
||||
if time.time() - time_of_last_run > 3:
|
||||
return
|
||||
else:
|
||||
if my_id:
|
||||
time_of_last_run = time.time()
|
||||
subprocess.call(["php", pathname+"/../../multiprocessing/.do_not_run/switch.php", "python "+my_id])
|
||||
time.sleep(.03)
|
||||
self.my_queue.task_done()
|
||||
|
||||
def main():
|
||||
global time_of_last_run
|
||||
time_of_last_run = time.time()
|
||||
|
||||
def signal_handler(signal, frame):
|
||||
sys.exit(0)
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
|
||||
if True:
|
||||
#spawn a pool of place worker threads
|
||||
for i in range(run_threads):
|
||||
p = queue_runner(my_queue)
|
||||
p.setDaemon(False)
|
||||
p.start()
|
||||
|
||||
#now load some arbitrary jobs into the queue
|
||||
run = 0
|
||||
finals = []
|
||||
groups = []
|
||||
s = name = ""
|
||||
for group in datas:
|
||||
time.sleep(.03)
|
||||
#start new groups using binaries.php, no need to check pynntp
|
||||
if group[1] == 0:
|
||||
run += 1
|
||||
my_queue.put("update_group_headers %s" % (group[0]))
|
||||
time.sleep(0.01)
|
||||
elif group[1] != 0:
|
||||
#only process if more that 20k headers available and skip the first 20k
|
||||
count = group[2] - group[1] - 20000
|
||||
#run small groups using binaries.php
|
||||
if count <= maxmssgs * 2:
|
||||
run += 1
|
||||
my_queue.put("update_group_headers %s" % (group[0]))
|
||||
#thread large groups using backfill.php
|
||||
else:
|
||||
my_queue.put("part_repair %s" % (group[0]))
|
||||
geteach = math.floor(count / maxmssgs)
|
||||
remaining = count - geteach * maxmssgs
|
||||
for loop in range(int(geteach)):
|
||||
run += 1
|
||||
my_queue.put("get_range binaries %s %s %s %s" % (group[0], group[1] + loop * maxmssgs + 1, group[1] + loop * maxmssgs + maxmssgs, run))
|
||||
run += 1
|
||||
my_queue.put("get_range binaries %s %s %s %s" % (group[0], group[1] + (loop + 1) * maxmssgs + 1, group[1] + (loop + 1) * maxmssgs + remaining + 1, run))
|
||||
groups.append(group[0])
|
||||
finals.append(int(group[2]))
|
||||
|
||||
my_queue.join()
|
||||
|
||||
print(bcolors.HEADER + "\nBinaries Safe Threaded Completed at {}".format(datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
print(bcolors.HEADER + "Running time: {}\n\n".format(str(datetime.timedelta(seconds=time.time() - start_time))) + bcolors.ENDC)
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
@@ -0,0 +1,119 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
from __future__ import print_function
|
||||
import sys, os, time
|
||||
import threading
|
||||
try:
|
||||
import queue
|
||||
except ImportError:
|
||||
import Queue as queue
|
||||
import subprocess
|
||||
import string
|
||||
import signal
|
||||
import datetime
|
||||
|
||||
import lib.info as info
|
||||
from lib.info import bcolors
|
||||
conf = info.readConfig()
|
||||
cur = info.connect()
|
||||
start_time = time.time()
|
||||
pathname = os.path.abspath(os.path.dirname(sys.argv[0]))
|
||||
|
||||
if len(sys.argv) == 1:
|
||||
print(bcolors.HEADER + "\nThis script will run update_binaries per group."
|
||||
"\nThis script can run on 1 group, an array of groups or all groups.\n"
|
||||
"\nEach group is processed in a single thread, for all groups. For example, 10 groups, 10 threads, upto max threads.\n"
|
||||
"\npython " + sys.argv[0] + " 155 ...: To run against groupid 155."
|
||||
"\npython " + sys.argv[0] + " '(155, 52)' ...: To run against groupid 155 and 52."
|
||||
"\npython " + sys.argv[0] + " alt.binaries.tv ...: To run against group alt.binaries.teevee."
|
||||
"\npython " + sys.argv[0] + " ...: To run against all active groups." + bcolors.ENDC)
|
||||
|
||||
print(bcolors.HEADER + "\nBinaries Threaded Started at {}".format(datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
|
||||
|
||||
|
||||
#get active groups
|
||||
if len(sys.argv) == 2:
|
||||
try:
|
||||
cur[0].execute("SELECT name FROM groups WHERE id IN " + sys.argv[1])
|
||||
datas = cur[0].fetchall()
|
||||
except:
|
||||
cur[0].execute("SELECT name FROM groups WHERE name = '" + sys.argv[1] + "'")
|
||||
datas = cur[0].fetchall()
|
||||
if len(datas) == 0:
|
||||
cur[0].execute("SELECT name FROM groups WHERE id = " + sys.argv[1])
|
||||
datas = cur[0].fetchall()
|
||||
if len(datas) == 0:
|
||||
print(bcolors.ERROR + "No Active Groups" + bcolors.ENDC)
|
||||
else:
|
||||
cur[0].execute("SELECT name FROM groups WHERE active = 1")
|
||||
datas = cur[0].fetchall()
|
||||
|
||||
if len(datas) == 0:
|
||||
print(bcolors.ERROR + "No Active Groups" + bcolors.ENDC)
|
||||
info.disconnect(cur[0], cur[1])
|
||||
sys.exit
|
||||
|
||||
cur[0].execute("SELECT value FROM tmux WHERE setting = 'binarythreads'")
|
||||
dbgrab = cur[0].fetchone()
|
||||
run_threads = int(dbgrab[0][0])
|
||||
|
||||
#close connection to mysql
|
||||
info.disconnect(cur[0], cur[1])
|
||||
|
||||
|
||||
my_queue = queue.Queue()
|
||||
time_of_last_run = time.time()
|
||||
|
||||
class queue_runner(threading.Thread):
|
||||
def __init__(self, my_queue):
|
||||
threading.Thread.__init__(self)
|
||||
self.my_queue = my_queue
|
||||
|
||||
def run(self):
|
||||
global time_of_last_run
|
||||
|
||||
while True:
|
||||
try:
|
||||
my_id = self.my_queue.get(True, .5)
|
||||
except:
|
||||
if time.time() - time_of_last_run > 3:
|
||||
return
|
||||
else:
|
||||
if my_id:
|
||||
time_of_last_run = time.time()
|
||||
subprocess.call(["php", pathname+"/../bin/update_binaries.php", ""+my_id])
|
||||
self.my_queue.task_done()
|
||||
|
||||
def main():
|
||||
global time_of_last_run
|
||||
time_of_last_run = time.time()
|
||||
|
||||
print(bcolors.HEADER + "We will be using a max of {} threads, a queue of {} groups".format(run_threads, "{:,}".format(len(datas))) + bcolors.ENDC)
|
||||
time.sleep(2)
|
||||
|
||||
def signal_handler(signal, frame):
|
||||
sys.exit(0)
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
|
||||
if True:
|
||||
#spawn a pool of worker threads
|
||||
for i in range(run_threads):
|
||||
p = queue_runner(my_queue)
|
||||
#p.setDaemon(False)
|
||||
p.start()
|
||||
|
||||
#now load some arbitrary jobs into the queue
|
||||
for gnames in datas:
|
||||
time.sleep(.03)
|
||||
my_queue.put(gnames[0])
|
||||
|
||||
my_queue.join()
|
||||
|
||||
print(bcolors.HEADER + "\nBinaries Threaded Completed at {}".format(datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
print(bcolors.HEADER + "Running time: {}\n\n".format(str(datetime.timedelta(seconds=time.time() - start_time))) + bcolors.ENDC)
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
@@ -0,0 +1,166 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
from __future__ import print_function
|
||||
import sys, os, time
|
||||
import threading
|
||||
try:
|
||||
import queue
|
||||
except ImportError:
|
||||
import Queue as queue
|
||||
import subprocess
|
||||
import string
|
||||
import signal
|
||||
import datetime
|
||||
|
||||
import lib.info as info
|
||||
from lib.info import bcolors
|
||||
conf = info.readConfig()
|
||||
cur = info.connect()
|
||||
start_time = time.time()
|
||||
pathname = os.path.abspath(os.path.dirname(sys.argv[0]))
|
||||
|
||||
if len(sys.argv) == 1:
|
||||
print(bcolors.ERROR + "\nAn argument is required\n\n"
|
||||
+ "python " + sys.argv[0] + " [md5, nfo, filename, par2] ...: To process all previously unprocessed releases, using [md5, nfo, filename, par2].\n"
|
||||
+ "python " + sys.argv[0] + " [nfo, filename, par2] prehashid ...: To process all releases not matched to prehashid, using [nfo, filename, par2].\n"
|
||||
+ "python " + sys.argv[0] + " nfo clean ...: To process all releases processed by filename, using nfo.\n"
|
||||
+ "python " + sys.argv[0] + " par2 clean ...: To process all releases processed by filename and nfo, using par2.\n"
|
||||
+ "python " + sys.argv[0] + " predbft clean ...: To process all releases using reverse match by PreDB title.\n"
|
||||
+ bcolors.ENDC)
|
||||
sys.exit()
|
||||
|
||||
if sys.argv[1] != "nfo" and sys.argv[1] != "filename" and sys.argv[1] != "md5" and sys.argv[1] != "par2" and sys.argv[1] != "predbft":
|
||||
print(bcolors.ERROR + "\n\An invalid argument was supplied\npostprocess_threaded.py [md5, nfo, filename, par2, predbft]\n" + bcolors.ENDC)
|
||||
sys.exit()
|
||||
|
||||
if len(sys.argv) == 3 and sys.argv[1] == "nfo" and sys.argv[2] == "clean":
|
||||
clean = " isrenamed = 0 AND proc_files = 1 "
|
||||
elif len(sys.argv) == 3 and sys.argv[1] == "par2" and sys.argv[2] == "clean":
|
||||
clean = " isrenamed = 0 AND proc_files = 1 AND proc_nfo = 1 "
|
||||
elif len(sys.argv) == 3 and sys.argv[1] == "nfo" and sys.argv[2] == "prehashid":
|
||||
clean = " prehashid = IS NULL "
|
||||
elif len(sys.argv) == 3 and sys.argv[1] == "par2" and sys.argv[2] == "prehashid":
|
||||
clean = " prehashid = IS NULL "
|
||||
elif len(sys.argv) == 3 and sys.argv[1] == "filename" and sys.argv[2] == "prehashid":
|
||||
clean = " prehashid = IS NULL "
|
||||
else:
|
||||
clean = " isrenamed = 0 "
|
||||
|
||||
print(bcolors.HEADER + "\nfixReleasesNames {} Threaded Started at {}".format(sys.argv[1],datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
|
||||
cur[0].execute("SELECT value FROM tmux WHERE setting = 'fixnamethreads'")
|
||||
run_threads = cur[0].fetchone()
|
||||
cur[0].execute("SELECT value FROM tmux WHERE setting = 'fixnamesperrun'")
|
||||
perrun = cur[0].fetchone()
|
||||
|
||||
datas = []
|
||||
maxtries = 0
|
||||
|
||||
if len(sys.argv) > 1 and sys.argv[1] == "nfo":
|
||||
run = "SELECT DISTINCT id AS releaseid FROM releases WHERE nzbstatus = 1 AND nfostatus = 1 AND proc_nfo = 0 AND" + clean + "ORDER BY postdate DESC LIMIT %s"
|
||||
cur[0].execute(run, (int(perrun[0]) * int(run_threads[0])))
|
||||
datas = cur[0].fetchall()
|
||||
elif len(sys.argv) > 1 and (sys.argv[1] == "filename"):
|
||||
run = "SELECT DISTINCT rel.id AS releaseid FROM releases rel INNER JOIN releasefiles relfiles ON (relfiles.releaseid = rel.id) WHERE categoryid IN (8010, 8020) AND proc_files = 0 AND" + clean + "ORDER BY postdate ASC LIMIT %s"
|
||||
cur[0].execute(run, (int(perrun[0]) * int(run_threads[0])))
|
||||
datas = cur[0].fetchall()
|
||||
elif len(sys.argv) > 1 and (sys.argv[1] == "md5"):
|
||||
while len(datas) == 0 and maxtries >= -5:
|
||||
run = "SELECT DISTINCT rel.id FROM releases rel INNER JOIN releasefiles rf ON rel.id = rf.releaseid WHERE rel.dehashstatus BETWEEN %s AND 0 AND (rel.ishashed = 1 OR rf.ishashed = 1) AND prehashid = 0 ORDER BY postdate ASC LIMIT %s"
|
||||
cur[0].execute(run, (maxtries, int(perrun[0])*int(run_threads[0])))
|
||||
datas = cur[0].fetchall()
|
||||
maxtries = maxtries - 1
|
||||
elif len(sys.argv) > 1 and (sys.argv[1] == "par2"):
|
||||
#This one does from oldest posts to newest posts, since nfo pp does same thing but newest to oldest
|
||||
run = "SELECT id AS releaseid, guid, groupid FROM releases WHERE categoryid IN (8010, 8020) AND proc_par2 = 0 AND" + clean + "ORDER BY postdate ASC LIMIT %s"
|
||||
cur[0].execute(run, (int(perrun[0]) * int(run_threads[0])))
|
||||
datas = cur[0].fetchall()
|
||||
elif len(sys.argv) > 1 and (sys.argv[1] == "predbft"):
|
||||
#This one does from oldest posts to newest posts since there are many other more efficient PreDB matching schemes
|
||||
run = "SELECT id AS preid FROM prehash WHERE LENGTH(title) >= 15 AND searched = 0 AND title NOT REGEXP '[\"\<\> ]' ORDER BY predate ASC LIMIT %s"
|
||||
cur[0].execute(run, (int(perrun[0])/10 * int(run_threads[0])))
|
||||
datas = cur[0].fetchall()
|
||||
|
||||
#close connection to mysql
|
||||
info.disconnect(cur[0], cur[1])
|
||||
|
||||
if not datas:
|
||||
print(bcolors.HEADER + "No Work to Process" + bcolors.ENDC)
|
||||
sys.exit()
|
||||
|
||||
my_queue = queue.Queue()
|
||||
time_of_last_run = time.time()
|
||||
|
||||
class queue_runner(threading.Thread):
|
||||
def __init__(self, my_queue):
|
||||
threading.Thread.__init__(self)
|
||||
self.my_queue = my_queue
|
||||
|
||||
def run(self):
|
||||
global time_of_last_run
|
||||
|
||||
while True:
|
||||
try:
|
||||
my_id = self.my_queue.get(True, 1)
|
||||
except:
|
||||
if time.time() - time_of_last_run > 3:
|
||||
return
|
||||
else:
|
||||
if my_id:
|
||||
time_of_last_run = time.time()
|
||||
subprocess.call(["php", pathname+"/../bin/fixreleasenames.php", ""+my_id])
|
||||
time.sleep(.03)
|
||||
self.my_queue.task_done()
|
||||
|
||||
def main():
|
||||
global time_of_last_run
|
||||
time_of_last_run = time.time()
|
||||
|
||||
if sys.argv[1] == 'md5':
|
||||
print(bcolors.HEADER + "We will be using a max of {} threads, a queue of {} {} releases. dehashstatus range {} to 0".format(run_threads[0], "{:,}".format(len(datas)), sys.argv[1], maxtries + 1) + bcolors.ENDC)
|
||||
else:
|
||||
print(bcolors.HEADER + "We will be using a max of {} threads, a queue of {} releases using {}".format(run_threads[0], "{:,}".format(len(datas)), sys.argv[1]) + bcolors.ENDC)
|
||||
time.sleep(2)
|
||||
|
||||
def signal_handler(signal, frame):
|
||||
sys.exit(0)
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
|
||||
if True:
|
||||
#spawn a pool of place worker threads
|
||||
for i in range(int(run_threads[0])):
|
||||
p = queue_runner(my_queue)
|
||||
p.setDaemon(False)
|
||||
p.start()
|
||||
|
||||
#now load some arbitrary jobs into the queue
|
||||
if sys.argv[1] == "nfo":
|
||||
for release in datas:
|
||||
time.sleep(.03)
|
||||
my_queue.put("%s %s" % ("nfo", release[0]))
|
||||
elif sys.argv[1] == "filename":
|
||||
for release in datas:
|
||||
time.sleep(.03)
|
||||
my_queue.put("%s %s" % ("filename", release[0]))
|
||||
elif sys.argv[1] == "md5":
|
||||
for release in datas:
|
||||
time.sleep(.03)
|
||||
my_queue.put("%s %s" % ("md5", release[0]))
|
||||
elif sys.argv[1] == "par2":
|
||||
for release in datas:
|
||||
time.sleep(.03)
|
||||
my_queue.put("%s %s %s %s" % ("par2", release[0], release[1], release[2]))
|
||||
elif sys.argv[1] == "predbft":
|
||||
for release in datas:
|
||||
time.sleep(.03)
|
||||
my_queue.put("%s %s" % ("predbft", release[0]))
|
||||
|
||||
my_queue.join()
|
||||
|
||||
print(bcolors.HEADER + "\nfixReleaseNames {} Threaded Completed at {}".format(sys.argv[1],datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
print(bcolors.HEADER + "Running time: {}\n\n".format(str(datetime.timedelta(seconds=time.time() - start_time))) + bcolors.ENDC)
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
@@ -0,0 +1,145 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import sys, os, time
|
||||
import threading
|
||||
try:
|
||||
import queue
|
||||
except ImportError:
|
||||
import Queue as queue
|
||||
import subprocess
|
||||
import string
|
||||
import signal
|
||||
import datetime
|
||||
|
||||
import lib.info as info
|
||||
from lib.info import bcolors
|
||||
conf = info.readConfig()
|
||||
cur = info.connect()
|
||||
start_time = time.time()
|
||||
pathname = os.path.abspath(os.path.dirname(sys.argv[0]))
|
||||
|
||||
if len(sys.argv) == 1:
|
||||
print(bcolors.ERROR + "\nAn argument is required\n\n"
|
||||
+ "python " + sys.argv[0] + " [md5, nfo, filename, par2, miscsorter, predbft] ...: To process all previously unprocessed releases, using [md5, nfo, filename, par2, miscsorter, predbft].\n"
|
||||
+ bcolors.ENDC)
|
||||
sys.exit()
|
||||
|
||||
if sys.argv[1] != "nfo" and sys.argv[1] != "filename" and sys.argv[1] != "md5" and sys.argv[1] != "par2" and sys.argv[1] != "miscsorter" and sys.argv[1] != "predbft":
|
||||
print(bcolors.ERROR + "\n\An invalid argument was supplied\ngroupfixrelnames_threaded.py [md5, nfo, filename, par2, miscsorter, predbft]\n" + bcolors.ENDC)
|
||||
sys.exit()
|
||||
|
||||
print(bcolors.HEADER + "\nfixReleaseNames {} Threaded Started at {}".format(sys.argv[1], datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
|
||||
datas = []
|
||||
limit = 0
|
||||
|
||||
join = ""
|
||||
where = ""
|
||||
datelimit = "AND DATEDIFF(NOW(), r.adddate) <= 7"
|
||||
groupby = "GROUP BY LEFT(r.guid, 1)"
|
||||
orderby = "ORDER BY LEFT(r.guid, 1) ASC"
|
||||
rowlimit = "LIMIT 100"
|
||||
|
||||
cur[0].execute("SELECT value FROM tmux WHERE setting = 'fixnamethreads'")
|
||||
run_threads = cur[0].fetchone()
|
||||
cur[0].execute("SELECT value FROM tmux WHERE setting = 'fixnamesperrun'")
|
||||
run_perrun = cur[0].fetchone()
|
||||
|
||||
threads = int(run_threads[0])
|
||||
if threads > 16:
|
||||
threads = 16
|
||||
maxperrun = int(run_perrun[0])
|
||||
|
||||
if sys.argv[1] == "md5":
|
||||
join = "LEFT OUTER JOIN releasefiles rf ON r.id = rf.releaseid AND rf.ishashed = 1"
|
||||
where = "r.ishashed = 1 AND r.dehashstatus BETWEEN -6 AND 0"
|
||||
elif sys.argv[1] == "nfo":
|
||||
where = "r.proc_nfo = 0 AND r.nfostatus = 1"
|
||||
elif sys.argv[1] == "filename":
|
||||
join = "INNER JOIN releasefiles rf ON r.id = rf.releaseid"
|
||||
where = "r.proc_files = 0"
|
||||
elif sys.argv[1] == "par2":
|
||||
where = "r.proc_par2 = 0"
|
||||
elif sys.argv[1] == "miscsorter":
|
||||
where = "r.nfostatus = 1 AND r.proc_sorter = 0 AND r.isrenamed = 0"
|
||||
elif sys.argv[1] == "predbft":
|
||||
where = "1=1"
|
||||
rowlimit = "LIMIT %s" % (threads)
|
||||
|
||||
cur[0].execute("SELECT DISTINCT LEFT(r.guid, 1), COUNT(*) AS count FROM releases r %s WHERE %s AND r.prehashid = 0 AND r.nzbstatus = 1 %s %s %s %s" % (join, where, datelimit, groupby, orderby, rowlimit))
|
||||
datas = cur[0].fetchall()
|
||||
|
||||
guids = int(len(datas))
|
||||
|
||||
#close connection to mysql
|
||||
info.disconnect(cur[0], cur[1])
|
||||
|
||||
if not datas:
|
||||
print(bcolors.HEADER + "No Work to Process" + bcolors.ENDC)
|
||||
sys.exit()
|
||||
|
||||
my_queue = queue.Queue()
|
||||
time_of_last_run = time.time()
|
||||
|
||||
class queue_runner(threading.Thread):
|
||||
def __init__(self, my_queue):
|
||||
threading.Thread.__init__(self)
|
||||
self.my_queue = my_queue
|
||||
|
||||
def run(self):
|
||||
global time_of_last_run
|
||||
|
||||
while True:
|
||||
try:
|
||||
my_id = self.my_queue.get(True, 1)
|
||||
except:
|
||||
if time.time() - time_of_last_run > 3:
|
||||
return
|
||||
else:
|
||||
if my_id:
|
||||
time_of_last_run = time.time()
|
||||
subprocess.call(["php", pathname+"/../bin/groupfixrelnames.php", ""+my_id])
|
||||
self.my_queue.task_done()
|
||||
|
||||
def main():
|
||||
global time_of_last_run
|
||||
time_of_last_run = time.time()
|
||||
|
||||
print(bcolors.HEADER + "We will be using a max of {} threads, a total of {} guid prefixes with a maximum of {} per thread.".format(threads, guids, maxperrun) + bcolors.ENDC)
|
||||
time.sleep(2)
|
||||
|
||||
def signal_handler(signal, frame):
|
||||
sys.exit(0)
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
|
||||
if True:
|
||||
#spawn a pool of place worker threads
|
||||
for i in range(threads):
|
||||
p = queue_runner(my_queue)
|
||||
p.setDaemon(False)
|
||||
p.start()
|
||||
|
||||
#now load some arbitrary jobs into the queue
|
||||
count = 0
|
||||
|
||||
for firstguid in datas:
|
||||
if count >= threads:
|
||||
count = 0
|
||||
count += 1
|
||||
if firstguid[1] < maxperrun:
|
||||
limit = firstguid[1]
|
||||
else:
|
||||
limit = maxperrun
|
||||
time.sleep(.03)
|
||||
if limit > 0:
|
||||
my_queue.put("%s %s %s %s" % (sys.argv[1], firstguid[0], limit, count))
|
||||
|
||||
my_queue.join()
|
||||
|
||||
print(bcolors.HEADER + "\nfixReleaseNames Per GUID Prefix Threaded Completed at {}".format(datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
print(bcolors.HEADER + "Running time: {}\n\n".format(str(datetime.timedelta(seconds=time.time() - start_time))) + bcolors.ENDC)
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
@@ -0,0 +1,112 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
from __future__ import print_function
|
||||
import sys, os, time
|
||||
import threading
|
||||
try:
|
||||
import queue
|
||||
except ImportError:
|
||||
import Queue as queue
|
||||
import subprocess
|
||||
import string
|
||||
import signal
|
||||
import datetime
|
||||
import lib.info as info
|
||||
from lib.info import bcolors
|
||||
conf = info.readConfig()
|
||||
cur = info.connect()
|
||||
start_time = time.time()
|
||||
pathname = os.path.abspath(os.path.dirname(sys.argv[0]))
|
||||
|
||||
print(bcolors.HEADER + "\nNZB Import Threaded Started at {}".format(datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
|
||||
#get values from db
|
||||
cur[0].execute("SELECT value FROM tmux WHERE setting = 'import'")
|
||||
use_true = cur[0].fetchone()
|
||||
cur[0].execute("SELECT (SELECT value FROM tmux WHERE setting = 'nzbthreads') AS a, (SELECT value FROM tmux WHERE setting = 'nzbs') AS b")
|
||||
dbgrab = cur[0].fetchall()
|
||||
run_threads = int(dbgrab[0][0])
|
||||
nzbs = dbgrab[0][1]
|
||||
|
||||
if int(use_true[0]) == 2 or ( len(sys.argv) >= 2 and sys.argv[1] == "true"):
|
||||
print(bcolors.HEADER + "We will be using filename as searchname" + bcolors.ENDC)
|
||||
print(bcolors.HEADER + "Sorting Folders in {}, be patient.".format(nzbs) + bcolors.ENDC)
|
||||
datas = [name for name in os.listdir(nzbs) if os.path.isdir(os.path.join(nzbs, name))]
|
||||
|
||||
#close connection to mysql
|
||||
info.disconnect(cur[0], cur[1])
|
||||
|
||||
my_queue = queue.Queue()
|
||||
time_of_last_run = time.time()
|
||||
|
||||
class queue_runner(threading.Thread):
|
||||
def __init__(self, my_queue):
|
||||
threading.Thread.__init__(self)
|
||||
self.my_queue = my_queue
|
||||
|
||||
def run(self):
|
||||
global time_of_last_run
|
||||
|
||||
while True:
|
||||
try:
|
||||
my_id = self.my_queue.get(True, 1)
|
||||
except:
|
||||
if time.time() - time_of_last_run > 3:
|
||||
return
|
||||
else:
|
||||
if my_id:
|
||||
time_of_last_run = time.time()
|
||||
subprocess.call(["php", pathname+"/../bin/nzb-import-new.php", ""+my_id])
|
||||
time.sleep(.03)
|
||||
self.my_queue.task_done()
|
||||
|
||||
def main(args):
|
||||
global time_of_last_run
|
||||
time_of_last_run = time.time()
|
||||
|
||||
if len(datas) != 0:
|
||||
print(bcolors.HEADER + "We will be using a max of {} threads, a queue of {} folders".format(run_threads, "{:,}".format(len(datas))) + bcolors.ENDC)
|
||||
else:
|
||||
print(bcolors.HEADER + "We will be using a max of {} threads, a queue of 1 folder".format(run_threads) + bcolors.ENDC)
|
||||
if int(use_true[0]) == 2 or ( len(sys.argv) >= 2 and sys.argv[1] == "true"):
|
||||
print(bcolors.HEADER + "We will be using filename as searchname" + bcolors.ENDC)
|
||||
time.sleep(2)
|
||||
|
||||
def signal_handler(signal, frame):
|
||||
sys.exit(0)
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
|
||||
if True:
|
||||
#spawn a pool of place worker threads
|
||||
for i in range(run_threads):
|
||||
p = queue_runner(my_queue)
|
||||
p.setDaemon(False)
|
||||
p.start()
|
||||
|
||||
#now load some arbitrary jobs into the queue
|
||||
if len(datas) != 0:
|
||||
if (int(use_true[0]) == 0 or int(use_true[0]) == 1) and len(sys.argv) == 1:
|
||||
for gnames in datas:
|
||||
time.sleep(.03)
|
||||
my_queue.put(os.path.join(nzbs,gnames))
|
||||
elif int(use_true[0]) == 2 or ( len(sys.argv) >= 2 and sys.argv[1] == "true"):
|
||||
for gnames in datas:
|
||||
time.sleep(.03)
|
||||
my_queue.put("%s %s" % (os.path.join(nzbs,gnames), "true"))
|
||||
if len(datas) == 0:
|
||||
if (int(use_true[0]) == 0 or int(use_true[0]) == 1) and len(sys.argv) == 1:
|
||||
time.sleep(.03)
|
||||
my_queue.put(nzbs)
|
||||
elif int(use_true[0]) == 2 or ( len(sys.argv) >= 2 and sys.argv[1] == "true"):
|
||||
time.sleep(.03)
|
||||
my_queue.put("%s %s" % (nzbs, "true"))
|
||||
|
||||
my_queue.join()
|
||||
|
||||
print(bcolors.HEADER + "\nNZB Import Threaded Completed at {}".format(datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
print(bcolors.HEADER + "Running time: {}\n\n".format(str(datetime.timedelta(seconds=time.time() - start_time))) + bcolors.ENDC)
|
||||
|
||||
if __name__ == '__main__':
|
||||
main(sys.argv[1:])
|
||||
@@ -0,0 +1,4 @@
|
||||
*.pyc
|
||||
*.pyo
|
||||
nntpproxy.conf
|
||||
nntpproxy_a.conf
|
||||
@@ -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()
|
||||
@@ -0,0 +1,16 @@
|
||||
{
|
||||
"usenet": {
|
||||
"host": "your.usenet.host.com",
|
||||
"port": 443,
|
||||
"username": "your_username",
|
||||
"password": "your_password",
|
||||
"use_ssl": true
|
||||
},
|
||||
"proxy": {
|
||||
"host": "localhost",
|
||||
"port": 9991
|
||||
},
|
||||
"pool": {
|
||||
"size": 5
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,12 @@
|
||||
# temp files
|
||||
~*
|
||||
.*.swp
|
||||
|
||||
# python compiled
|
||||
*.pyc
|
||||
|
||||
# setuptools stuff
|
||||
build/
|
||||
dist/
|
||||
MANIFEST
|
||||
pynntp.egg-info/
|
||||
@@ -0,0 +1,674 @@
|
||||
GNU GENERAL PUBLIC LICENSE
|
||||
Version 3, 29 June 2007
|
||||
|
||||
Copyright (C) 2007 Free Software Foundation, Inc. <http://fsf.org/>
|
||||
Everyone is permitted to copy and distribute verbatim copies
|
||||
of this license document, but changing it is not allowed.
|
||||
|
||||
Preamble
|
||||
|
||||
The GNU General Public License is a free, copyleft license for
|
||||
software and other kinds of works.
|
||||
|
||||
The licenses for most software and other practical works are designed
|
||||
to take away your freedom to share and change the works. By contrast,
|
||||
the GNU General Public License is intended to guarantee your freedom to
|
||||
share and change all versions of a program--to make sure it remains free
|
||||
software for all its users. We, the Free Software Foundation, use the
|
||||
GNU General Public License for most of our software; it applies also to
|
||||
any other work released this way by its authors. You can apply it to
|
||||
your programs, too.
|
||||
|
||||
When we speak of free software, we are referring to freedom, not
|
||||
price. Our General Public Licenses are designed to make sure that you
|
||||
have the freedom to distribute copies of free software (and charge for
|
||||
them if you wish), that you receive source code or can get it if you
|
||||
want it, that you can change the software or use pieces of it in new
|
||||
free programs, and that you know you can do these things.
|
||||
|
||||
To protect your rights, we need to prevent others from denying you
|
||||
these rights or asking you to surrender the rights. Therefore, you have
|
||||
certain responsibilities if you distribute copies of the software, or if
|
||||
you modify it: responsibilities to respect the freedom of others.
|
||||
|
||||
For example, if you distribute copies of such a program, whether
|
||||
gratis or for a fee, you must pass on to the recipients the same
|
||||
freedoms that you received. You must make sure that they, too, receive
|
||||
or can get the source code. And you must show them these terms so they
|
||||
know their rights.
|
||||
|
||||
Developers that use the GNU GPL protect your rights with two steps:
|
||||
(1) assert copyright on the software, and (2) offer you this License
|
||||
giving you legal permission to copy, distribute and/or modify it.
|
||||
|
||||
For the developers' and authors' protection, the GPL clearly explains
|
||||
that there is no warranty for this free software. For both users' and
|
||||
authors' sake, the GPL requires that modified versions be marked as
|
||||
changed, so that their problems will not be attributed erroneously to
|
||||
authors of previous versions.
|
||||
|
||||
Some devices are designed to deny users access to install or run
|
||||
modified versions of the software inside them, although the manufacturer
|
||||
can do so. This is fundamentally incompatible with the aim of
|
||||
protecting users' freedom to change the software. The systematic
|
||||
pattern of such abuse occurs in the area of products for individuals to
|
||||
use, which is precisely where it is most unacceptable. Therefore, we
|
||||
have designed this version of the GPL to prohibit the practice for those
|
||||
products. If such problems arise substantially in other domains, we
|
||||
stand ready to extend this provision to those domains in future versions
|
||||
of the GPL, as needed to protect the freedom of users.
|
||||
|
||||
Finally, every program is threatened constantly by software patents.
|
||||
States should not allow patents to restrict development and use of
|
||||
software on general-purpose computers, but in those that do, we wish to
|
||||
avoid the special danger that patents applied to a free program could
|
||||
make it effectively proprietary. To prevent this, the GPL assures that
|
||||
patents cannot be used to render the program non-free.
|
||||
|
||||
The precise terms and conditions for copying, distribution and
|
||||
modification follow.
|
||||
|
||||
TERMS AND CONDITIONS
|
||||
|
||||
0. Definitions.
|
||||
|
||||
"This License" refers to version 3 of the GNU General Public License.
|
||||
|
||||
"Copyright" also means copyright-like laws that apply to other kinds of
|
||||
works, such as semiconductor masks.
|
||||
|
||||
"The Program" refers to any copyrightable work licensed under this
|
||||
License. Each licensee is addressed as "you". "Licensees" and
|
||||
"recipients" may be individuals or organizations.
|
||||
|
||||
To "modify" a work means to copy from or adapt all or part of the work
|
||||
in a fashion requiring copyright permission, other than the making of an
|
||||
exact copy. The resulting work is called a "modified version" of the
|
||||
earlier work or a work "based on" the earlier work.
|
||||
|
||||
A "covered work" means either the unmodified Program or a work based
|
||||
on the Program.
|
||||
|
||||
To "propagate" a work means to do anything with it that, without
|
||||
permission, would make you directly or secondarily liable for
|
||||
infringement under applicable copyright law, except executing it on a
|
||||
computer or modifying a private copy. Propagation includes copying,
|
||||
distribution (with or without modification), making available to the
|
||||
public, and in some countries other activities as well.
|
||||
|
||||
To "convey" a work means any kind of propagation that enables other
|
||||
parties to make or receive copies. Mere interaction with a user through
|
||||
a computer network, with no transfer of a copy, is not conveying.
|
||||
|
||||
An interactive user interface displays "Appropriate Legal Notices"
|
||||
to the extent that it includes a convenient and prominently visible
|
||||
feature that (1) displays an appropriate copyright notice, and (2)
|
||||
tells the user that there is no warranty for the work (except to the
|
||||
extent that warranties are provided), that licensees may convey the
|
||||
work under this License, and how to view a copy of this License. If
|
||||
the interface presents a list of user commands or options, such as a
|
||||
menu, a prominent item in the list meets this criterion.
|
||||
|
||||
1. Source Code.
|
||||
|
||||
The "source code" for a work means the preferred form of the work
|
||||
for making modifications to it. "Object code" means any non-source
|
||||
form of a work.
|
||||
|
||||
A "Standard Interface" means an interface that either is an official
|
||||
standard defined by a recognized standards body, or, in the case of
|
||||
interfaces specified for a particular programming language, one that
|
||||
is widely used among developers working in that language.
|
||||
|
||||
The "System Libraries" of an executable work include anything, other
|
||||
than the work as a whole, that (a) is included in the normal form of
|
||||
packaging a Major Component, but which is not part of that Major
|
||||
Component, and (b) serves only to enable use of the work with that
|
||||
Major Component, or to implement a Standard Interface for which an
|
||||
implementation is available to the public in source code form. A
|
||||
"Major Component", in this context, means a major essential component
|
||||
(kernel, window system, and so on) of the specific operating system
|
||||
(if any) on which the executable work runs, or a compiler used to
|
||||
produce the work, or an object code interpreter used to run it.
|
||||
|
||||
The "Corresponding Source" for a work in object code form means all
|
||||
the source code needed to generate, install, and (for an executable
|
||||
work) run the object code and to modify the work, including scripts to
|
||||
control those activities. However, it does not include the work's
|
||||
System Libraries, or general-purpose tools or generally available free
|
||||
programs which are used unmodified in performing those activities but
|
||||
which are not part of the work. For example, Corresponding Source
|
||||
includes interface definition files associated with source files for
|
||||
the work, and the source code for shared libraries and dynamically
|
||||
linked subprograms that the work is specifically designed to require,
|
||||
such as by intimate data communication or control flow between those
|
||||
subprograms and other parts of the work.
|
||||
|
||||
The Corresponding Source need not include anything that users
|
||||
can regenerate automatically from other parts of the Corresponding
|
||||
Source.
|
||||
|
||||
The Corresponding Source for a work in source code form is that
|
||||
same work.
|
||||
|
||||
2. Basic Permissions.
|
||||
|
||||
All rights granted under this License are granted for the term of
|
||||
copyright on the Program, and are irrevocable provided the stated
|
||||
conditions are met. This License explicitly affirms your unlimited
|
||||
permission to run the unmodified Program. The output from running a
|
||||
covered work is covered by this License only if the output, given its
|
||||
content, constitutes a covered work. This License acknowledges your
|
||||
rights of fair use or other equivalent, as provided by copyright law.
|
||||
|
||||
You may make, run and propagate covered works that you do not
|
||||
convey, without conditions so long as your license otherwise remains
|
||||
in force. You may convey covered works to others for the sole purpose
|
||||
of having them make modifications exclusively for you, or provide you
|
||||
with facilities for running those works, provided that you comply with
|
||||
the terms of this License in conveying all material for which you do
|
||||
not control copyright. Those thus making or running the covered works
|
||||
for you must do so exclusively on your behalf, under your direction
|
||||
and control, on terms that prohibit them from making any copies of
|
||||
your copyrighted material outside their relationship with you.
|
||||
|
||||
Conveying under any other circumstances is permitted solely under
|
||||
the conditions stated below. Sublicensing is not allowed; section 10
|
||||
makes it unnecessary.
|
||||
|
||||
3. Protecting Users' Legal Rights From Anti-Circumvention Law.
|
||||
|
||||
No covered work shall be deemed part of an effective technological
|
||||
measure under any applicable law fulfilling obligations under article
|
||||
11 of the WIPO copyright treaty adopted on 20 December 1996, or
|
||||
similar laws prohibiting or restricting circumvention of such
|
||||
measures.
|
||||
|
||||
When you convey a covered work, you waive any legal power to forbid
|
||||
circumvention of technological measures to the extent such circumvention
|
||||
is effected by exercising rights under this License with respect to
|
||||
the covered work, and you disclaim any intention to limit operation or
|
||||
modification of the work as a means of enforcing, against the work's
|
||||
users, your or third parties' legal rights to forbid circumvention of
|
||||
technological measures.
|
||||
|
||||
4. Conveying Verbatim Copies.
|
||||
|
||||
You may convey verbatim copies of the Program's source code as you
|
||||
receive it, in any medium, provided that you conspicuously and
|
||||
appropriately publish on each copy an appropriate copyright notice;
|
||||
keep intact all notices stating that this License and any
|
||||
non-permissive terms added in accord with section 7 apply to the code;
|
||||
keep intact all notices of the absence of any warranty; and give all
|
||||
recipients a copy of this License along with the Program.
|
||||
|
||||
You may charge any price or no price for each copy that you convey,
|
||||
and you may offer support or warranty protection for a fee.
|
||||
|
||||
5. Conveying Modified Source Versions.
|
||||
|
||||
You may convey a work based on the Program, or the modifications to
|
||||
produce it from the Program, in the form of source code under the
|
||||
terms of section 4, provided that you also meet all of these conditions:
|
||||
|
||||
a) The work must carry prominent notices stating that you modified
|
||||
it, and giving a relevant date.
|
||||
|
||||
b) The work must carry prominent notices stating that it is
|
||||
released under this License and any conditions added under section
|
||||
7. This requirement modifies the requirement in section 4 to
|
||||
"keep intact all notices".
|
||||
|
||||
c) You must license the entire work, as a whole, under this
|
||||
License to anyone who comes into possession of a copy. This
|
||||
License will therefore apply, along with any applicable section 7
|
||||
additional terms, to the whole of the work, and all its parts,
|
||||
regardless of how they are packaged. This License gives no
|
||||
permission to license the work in any other way, but it does not
|
||||
invalidate such permission if you have separately received it.
|
||||
|
||||
d) If the work has interactive user interfaces, each must display
|
||||
Appropriate Legal Notices; however, if the Program has interactive
|
||||
interfaces that do not display Appropriate Legal Notices, your
|
||||
work need not make them do so.
|
||||
|
||||
A compilation of a covered work with other separate and independent
|
||||
works, which are not by their nature extensions of the covered work,
|
||||
and which are not combined with it such as to form a larger program,
|
||||
in or on a volume of a storage or distribution medium, is called an
|
||||
"aggregate" if the compilation and its resulting copyright are not
|
||||
used to limit the access or legal rights of the compilation's users
|
||||
beyond what the individual works permit. Inclusion of a covered work
|
||||
in an aggregate does not cause this License to apply to the other
|
||||
parts of the aggregate.
|
||||
|
||||
6. Conveying Non-Source Forms.
|
||||
|
||||
You may convey a covered work in object code form under the terms
|
||||
of sections 4 and 5, provided that you also convey the
|
||||
machine-readable Corresponding Source under the terms of this License,
|
||||
in one of these ways:
|
||||
|
||||
a) Convey the object code in, or embodied in, a physical product
|
||||
(including a physical distribution medium), accompanied by the
|
||||
Corresponding Source fixed on a durable physical medium
|
||||
customarily used for software interchange.
|
||||
|
||||
b) Convey the object code in, or embodied in, a physical product
|
||||
(including a physical distribution medium), accompanied by a
|
||||
written offer, valid for at least three years and valid for as
|
||||
long as you offer spare parts or customer support for that product
|
||||
model, to give anyone who possesses the object code either (1) a
|
||||
copy of the Corresponding Source for all the software in the
|
||||
product that is covered by this License, on a durable physical
|
||||
medium customarily used for software interchange, for a price no
|
||||
more than your reasonable cost of physically performing this
|
||||
conveying of source, or (2) access to copy the
|
||||
Corresponding Source from a network server at no charge.
|
||||
|
||||
c) Convey individual copies of the object code with a copy of the
|
||||
written offer to provide the Corresponding Source. This
|
||||
alternative is allowed only occasionally and noncommercially, and
|
||||
only if you received the object code with such an offer, in accord
|
||||
with subsection 6b.
|
||||
|
||||
d) Convey the object code by offering access from a designated
|
||||
place (gratis or for a charge), and offer equivalent access to the
|
||||
Corresponding Source in the same way through the same place at no
|
||||
further charge. You need not require recipients to copy the
|
||||
Corresponding Source along with the object code. If the place to
|
||||
copy the object code is a network server, the Corresponding Source
|
||||
may be on a different server (operated by you or a third party)
|
||||
that supports equivalent copying facilities, provided you maintain
|
||||
clear directions next to the object code saying where to find the
|
||||
Corresponding Source. Regardless of what server hosts the
|
||||
Corresponding Source, you remain obligated to ensure that it is
|
||||
available for as long as needed to satisfy these requirements.
|
||||
|
||||
e) Convey the object code using peer-to-peer transmission, provided
|
||||
you inform other peers where the object code and Corresponding
|
||||
Source of the work are being offered to the general public at no
|
||||
charge under subsection 6d.
|
||||
|
||||
A separable portion of the object code, whose source code is excluded
|
||||
from the Corresponding Source as a System Library, need not be
|
||||
included in conveying the object code work.
|
||||
|
||||
A "User Product" is either (1) a "consumer product", which means any
|
||||
tangible personal property which is normally used for personal, family,
|
||||
or household purposes, or (2) anything designed or sold for incorporation
|
||||
into a dwelling. In determining whether a product is a consumer product,
|
||||
doubtful cases shall be resolved in favor of coverage. For a particular
|
||||
product received by a particular user, "normally used" refers to a
|
||||
typical or common use of that class of product, regardless of the status
|
||||
of the particular user or of the way in which the particular user
|
||||
actually uses, or expects or is expected to use, the product. A product
|
||||
is a consumer product regardless of whether the product has substantial
|
||||
commercial, industrial or non-consumer uses, unless such uses represent
|
||||
the only significant mode of use of the product.
|
||||
|
||||
"Installation Information" for a User Product means any methods,
|
||||
procedures, authorization keys, or other information required to install
|
||||
and execute modified versions of a covered work in that User Product from
|
||||
a modified version of its Corresponding Source. The information must
|
||||
suffice to ensure that the continued functioning of the modified object
|
||||
code is in no case prevented or interfered with solely because
|
||||
modification has been made.
|
||||
|
||||
If you convey an object code work under this section in, or with, or
|
||||
specifically for use in, a User Product, and the conveying occurs as
|
||||
part of a transaction in which the right of possession and use of the
|
||||
User Product is transferred to the recipient in perpetuity or for a
|
||||
fixed term (regardless of how the transaction is characterized), the
|
||||
Corresponding Source conveyed under this section must be accompanied
|
||||
by the Installation Information. But this requirement does not apply
|
||||
if neither you nor any third party retains the ability to install
|
||||
modified object code on the User Product (for example, the work has
|
||||
been installed in ROM).
|
||||
|
||||
The requirement to provide Installation Information does not include a
|
||||
requirement to continue to provide support service, warranty, or updates
|
||||
for a work that has been modified or installed by the recipient, or for
|
||||
the User Product in which it has been modified or installed. Access to a
|
||||
network may be denied when the modification itself materially and
|
||||
adversely affects the operation of the network or violates the rules and
|
||||
protocols for communication across the network.
|
||||
|
||||
Corresponding Source conveyed, and Installation Information provided,
|
||||
in accord with this section must be in a format that is publicly
|
||||
documented (and with an implementation available to the public in
|
||||
source code form), and must require no special password or key for
|
||||
unpacking, reading or copying.
|
||||
|
||||
7. Additional Terms.
|
||||
|
||||
"Additional permissions" are terms that supplement the terms of this
|
||||
License by making exceptions from one or more of its conditions.
|
||||
Additional permissions that are applicable to the entire Program shall
|
||||
be treated as though they were included in this License, to the extent
|
||||
that they are valid under applicable law. If additional permissions
|
||||
apply only to part of the Program, that part may be used separately
|
||||
under those permissions, but the entire Program remains governed by
|
||||
this License without regard to the additional permissions.
|
||||
|
||||
When you convey a copy of a covered work, you may at your option
|
||||
remove any additional permissions from that copy, or from any part of
|
||||
it. (Additional permissions may be written to require their own
|
||||
removal in certain cases when you modify the work.) You may place
|
||||
additional permissions on material, added by you to a covered work,
|
||||
for which you have or can give appropriate copyright permission.
|
||||
|
||||
Notwithstanding any other provision of this License, for material you
|
||||
add to a covered work, you may (if authorized by the copyright holders of
|
||||
that material) supplement the terms of this License with terms:
|
||||
|
||||
a) Disclaiming warranty or limiting liability differently from the
|
||||
terms of sections 15 and 16 of this License; or
|
||||
|
||||
b) Requiring preservation of specified reasonable legal notices or
|
||||
author attributions in that material or in the Appropriate Legal
|
||||
Notices displayed by works containing it; or
|
||||
|
||||
c) Prohibiting misrepresentation of the origin of that material, or
|
||||
requiring that modified versions of such material be marked in
|
||||
reasonable ways as different from the original version; or
|
||||
|
||||
d) Limiting the use for publicity purposes of names of licensors or
|
||||
authors of the material; or
|
||||
|
||||
e) Declining to grant rights under trademark law for use of some
|
||||
trade names, trademarks, or service marks; or
|
||||
|
||||
f) Requiring indemnification of licensors and authors of that
|
||||
material by anyone who conveys the material (or modified versions of
|
||||
it) with contractual assumptions of liability to the recipient, for
|
||||
any liability that these contractual assumptions directly impose on
|
||||
those licensors and authors.
|
||||
|
||||
All other non-permissive additional terms are considered "further
|
||||
restrictions" within the meaning of section 10. If the Program as you
|
||||
received it, or any part of it, contains a notice stating that it is
|
||||
governed by this License along with a term that is a further
|
||||
restriction, you may remove that term. If a license document contains
|
||||
a further restriction but permits relicensing or conveying under this
|
||||
License, you may add to a covered work material governed by the terms
|
||||
of that license document, provided that the further restriction does
|
||||
not survive such relicensing or conveying.
|
||||
|
||||
If you add terms to a covered work in accord with this section, you
|
||||
must place, in the relevant source files, a statement of the
|
||||
additional terms that apply to those files, or a notice indicating
|
||||
where to find the applicable terms.
|
||||
|
||||
Additional terms, permissive or non-permissive, may be stated in the
|
||||
form of a separately written license, or stated as exceptions;
|
||||
the above requirements apply either way.
|
||||
|
||||
8. Termination.
|
||||
|
||||
You may not propagate or modify a covered work except as expressly
|
||||
provided under this License. Any attempt otherwise to propagate or
|
||||
modify it is void, and will automatically terminate your rights under
|
||||
this License (including any patent licenses granted under the third
|
||||
paragraph of section 11).
|
||||
|
||||
However, if you cease all violation of this License, then your
|
||||
license from a particular copyright holder is reinstated (a)
|
||||
provisionally, unless and until the copyright holder explicitly and
|
||||
finally terminates your license, and (b) permanently, if the copyright
|
||||
holder fails to notify you of the violation by some reasonable means
|
||||
prior to 60 days after the cessation.
|
||||
|
||||
Moreover, your license from a particular copyright holder is
|
||||
reinstated permanently if the copyright holder notifies you of the
|
||||
violation by some reasonable means, this is the first time you have
|
||||
received notice of violation of this License (for any work) from that
|
||||
copyright holder, and you cure the violation prior to 30 days after
|
||||
your receipt of the notice.
|
||||
|
||||
Termination of your rights under this section does not terminate the
|
||||
licenses of parties who have received copies or rights from you under
|
||||
this License. If your rights have been terminated and not permanently
|
||||
reinstated, you do not qualify to receive new licenses for the same
|
||||
material under section 10.
|
||||
|
||||
9. Acceptance Not Required for Having Copies.
|
||||
|
||||
You are not required to accept this License in order to receive or
|
||||
run a copy of the Program. Ancillary propagation of a covered work
|
||||
occurring solely as a consequence of using peer-to-peer transmission
|
||||
to receive a copy likewise does not require acceptance. However,
|
||||
nothing other than this License grants you permission to propagate or
|
||||
modify any covered work. These actions infringe copyright if you do
|
||||
not accept this License. Therefore, by modifying or propagating a
|
||||
covered work, you indicate your acceptance of this License to do so.
|
||||
|
||||
10. Automatic Licensing of Downstream Recipients.
|
||||
|
||||
Each time you convey a covered work, the recipient automatically
|
||||
receives a license from the original licensors, to run, modify and
|
||||
propagate that work, subject to this License. You are not responsible
|
||||
for enforcing compliance by third parties with this License.
|
||||
|
||||
An "entity transaction" is a transaction transferring control of an
|
||||
organization, or substantially all assets of one, or subdividing an
|
||||
organization, or merging organizations. If propagation of a covered
|
||||
work results from an entity transaction, each party to that
|
||||
transaction who receives a copy of the work also receives whatever
|
||||
licenses to the work the party's predecessor in interest had or could
|
||||
give under the previous paragraph, plus a right to possession of the
|
||||
Corresponding Source of the work from the predecessor in interest, if
|
||||
the predecessor has it or can get it with reasonable efforts.
|
||||
|
||||
You may not impose any further restrictions on the exercise of the
|
||||
rights granted or affirmed under this License. For example, you may
|
||||
not impose a license fee, royalty, or other charge for exercise of
|
||||
rights granted under this License, and you may not initiate litigation
|
||||
(including a cross-claim or counterclaim in a lawsuit) alleging that
|
||||
any patent claim is infringed by making, using, selling, offering for
|
||||
sale, or importing the Program or any portion of it.
|
||||
|
||||
11. Patents.
|
||||
|
||||
A "contributor" is a copyright holder who authorizes use under this
|
||||
License of the Program or a work on which the Program is based. The
|
||||
work thus licensed is called the contributor's "contributor version".
|
||||
|
||||
A contributor's "essential patent claims" are all patent claims
|
||||
owned or controlled by the contributor, whether already acquired or
|
||||
hereafter acquired, that would be infringed by some manner, permitted
|
||||
by this License, of making, using, or selling its contributor version,
|
||||
but do not include claims that would be infringed only as a
|
||||
consequence of further modification of the contributor version. For
|
||||
purposes of this definition, "control" includes the right to grant
|
||||
patent sublicenses in a manner consistent with the requirements of
|
||||
this License.
|
||||
|
||||
Each contributor grants you a non-exclusive, worldwide, royalty-free
|
||||
patent license under the contributor's essential patent claims, to
|
||||
make, use, sell, offer for sale, import and otherwise run, modify and
|
||||
propagate the contents of its contributor version.
|
||||
|
||||
In the following three paragraphs, a "patent license" is any express
|
||||
agreement or commitment, however denominated, not to enforce a patent
|
||||
(such as an express permission to practice a patent or covenant not to
|
||||
sue for patent infringement). To "grant" such a patent license to a
|
||||
party means to make such an agreement or commitment not to enforce a
|
||||
patent against the party.
|
||||
|
||||
If you convey a covered work, knowingly relying on a patent license,
|
||||
and the Corresponding Source of the work is not available for anyone
|
||||
to copy, free of charge and under the terms of this License, through a
|
||||
publicly available network server or other readily accessible means,
|
||||
then you must either (1) cause the Corresponding Source to be so
|
||||
available, or (2) arrange to deprive yourself of the benefit of the
|
||||
patent license for this particular work, or (3) arrange, in a manner
|
||||
consistent with the requirements of this License, to extend the patent
|
||||
license to downstream recipients. "Knowingly relying" means you have
|
||||
actual knowledge that, but for the patent license, your conveying the
|
||||
covered work in a country, or your recipient's use of the covered work
|
||||
in a country, would infringe one or more identifiable patents in that
|
||||
country that you have reason to believe are valid.
|
||||
|
||||
If, pursuant to or in connection with a single transaction or
|
||||
arrangement, you convey, or propagate by procuring conveyance of, a
|
||||
covered work, and grant a patent license to some of the parties
|
||||
receiving the covered work authorizing them to use, propagate, modify
|
||||
or convey a specific copy of the covered work, then the patent license
|
||||
you grant is automatically extended to all recipients of the covered
|
||||
work and works based on it.
|
||||
|
||||
A patent license is "discriminatory" if it does not include within
|
||||
the scope of its coverage, prohibits the exercise of, or is
|
||||
conditioned on the non-exercise of one or more of the rights that are
|
||||
specifically granted under this License. You may not convey a covered
|
||||
work if you are a party to an arrangement with a third party that is
|
||||
in the business of distributing software, under which you make payment
|
||||
to the third party based on the extent of your activity of conveying
|
||||
the work, and under which the third party grants, to any of the
|
||||
parties who would receive the covered work from you, a discriminatory
|
||||
patent license (a) in connection with copies of the covered work
|
||||
conveyed by you (or copies made from those copies), or (b) primarily
|
||||
for and in connection with specific products or compilations that
|
||||
contain the covered work, unless you entered into that arrangement,
|
||||
or that patent license was granted, prior to 28 March 2007.
|
||||
|
||||
Nothing in this License shall be construed as excluding or limiting
|
||||
any implied license or other defenses to infringement that may
|
||||
otherwise be available to you under applicable patent law.
|
||||
|
||||
12. No Surrender of Others' Freedom.
|
||||
|
||||
If conditions are imposed on you (whether by court order, agreement or
|
||||
otherwise) that contradict the conditions of this License, they do not
|
||||
excuse you from the conditions of this License. If you cannot convey a
|
||||
covered work so as to satisfy simultaneously your obligations under this
|
||||
License and any other pertinent obligations, then as a consequence you may
|
||||
not convey it at all. For example, if you agree to terms that obligate you
|
||||
to collect a royalty for further conveying from those to whom you convey
|
||||
the Program, the only way you could satisfy both those terms and this
|
||||
License would be to refrain entirely from conveying the Program.
|
||||
|
||||
13. Use with the GNU Affero General Public License.
|
||||
|
||||
Notwithstanding any other provision of this License, you have
|
||||
permission to link or combine any covered work with a work licensed
|
||||
under version 3 of the GNU Affero General Public License into a single
|
||||
combined work, and to convey the resulting work. The terms of this
|
||||
License will continue to apply to the part which is the covered work,
|
||||
but the special requirements of the GNU Affero General Public License,
|
||||
section 13, concerning interaction through a network will apply to the
|
||||
combination as such.
|
||||
|
||||
14. Revised Versions of this License.
|
||||
|
||||
The Free Software Foundation may publish revised and/or new versions of
|
||||
the GNU General Public License from time to time. Such new versions will
|
||||
be similar in spirit to the present version, but may differ in detail to
|
||||
address new problems or concerns.
|
||||
|
||||
Each version is given a distinguishing version number. If the
|
||||
Program specifies that a certain numbered version of the GNU General
|
||||
Public License "or any later version" applies to it, you have the
|
||||
option of following the terms and conditions either of that numbered
|
||||
version or of any later version published by the Free Software
|
||||
Foundation. If the Program does not specify a version number of the
|
||||
GNU General Public License, you may choose any version ever published
|
||||
by the Free Software Foundation.
|
||||
|
||||
If the Program specifies that a proxy can decide which future
|
||||
versions of the GNU General Public License can be used, that proxy's
|
||||
public statement of acceptance of a version permanently authorizes you
|
||||
to choose that version for the Program.
|
||||
|
||||
Later license versions may give you additional or different
|
||||
permissions. However, no additional obligations are imposed on any
|
||||
author or copyright holder as a result of your choosing to follow a
|
||||
later version.
|
||||
|
||||
15. Disclaimer of Warranty.
|
||||
|
||||
THERE IS NO WARRANTY FOR THE PROGRAM, TO THE EXTENT PERMITTED BY
|
||||
APPLICABLE LAW. EXCEPT WHEN OTHERWISE STATED IN WRITING THE COPYRIGHT
|
||||
HOLDERS AND/OR OTHER PARTIES PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY
|
||||
OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING, BUT NOT LIMITED TO,
|
||||
THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
||||
PURPOSE. THE ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE OF THE PROGRAM
|
||||
IS WITH YOU. SHOULD THE PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF
|
||||
ALL NECESSARY SERVICING, REPAIR OR CORRECTION.
|
||||
|
||||
16. Limitation of Liability.
|
||||
|
||||
IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING
|
||||
WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MODIFIES AND/OR CONVEYS
|
||||
THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES, INCLUDING ANY
|
||||
GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING OUT OF THE
|
||||
USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED TO LOSS OF
|
||||
DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY YOU OR THIRD
|
||||
PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER PROGRAMS),
|
||||
EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE POSSIBILITY OF
|
||||
SUCH DAMAGES.
|
||||
|
||||
17. Interpretation of Sections 15 and 16.
|
||||
|
||||
If the disclaimer of warranty and limitation of liability provided
|
||||
above cannot be given local legal effect according to their terms,
|
||||
reviewing courts shall apply local law that most closely approximates
|
||||
an absolute waiver of all civil liability in connection with the
|
||||
Program, unless a warranty or assumption of liability accompanies a
|
||||
copy of the Program in return for a fee.
|
||||
|
||||
END OF TERMS AND CONDITIONS
|
||||
|
||||
How to Apply These Terms to Your New Programs
|
||||
|
||||
If you develop a new program, and you want it to be of the greatest
|
||||
possible use to the public, the best way to achieve this is to make it
|
||||
free software which everyone can redistribute and change under these terms.
|
||||
|
||||
To do so, attach the following notices to the program. It is safest
|
||||
to attach them to the start of each source file to most effectively
|
||||
state the exclusion of warranty; and each file should have at least
|
||||
the "copyright" line and a pointer to where the full notice is found.
|
||||
|
||||
<one line to give the program's name and a brief idea of what it does.>
|
||||
Copyright (C) <year> <name of author>
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
Also add information on how to contact you by electronic and paper mail.
|
||||
|
||||
If the program does terminal interaction, make it output a short
|
||||
notice like this when it starts in an interactive mode:
|
||||
|
||||
<program> Copyright (C) <year> <name of author>
|
||||
This program comes with ABSOLUTELY NO WARRANTY; for details type `show w'.
|
||||
This is free software, and you are welcome to redistribute it
|
||||
under certain conditions; type `show c' for details.
|
||||
|
||||
The hypothetical commands `show w' and `show c' should show the appropriate
|
||||
parts of the General Public License. Of course, your program's commands
|
||||
might be different; for a GUI interface, you would use an "about box".
|
||||
|
||||
You should also get your employer (if you work as a programmer) or school,
|
||||
if any, to sign a "copyright disclaimer" for the program, if necessary.
|
||||
For more information on this, and how to apply and follow the GNU GPL, see
|
||||
<http://www.gnu.org/licenses/>.
|
||||
|
||||
The GNU General Public License does not permit incorporating your program
|
||||
into proprietary programs. If your program is a subroutine library, you
|
||||
may consider it more useful to permit linking proprietary applications with
|
||||
the library. If this is what you want to do, use the GNU Lesser General
|
||||
Public License instead of this License. But first, please read
|
||||
<http://www.gnu.org/philosophy/why-not-lgpl.html>.
|
||||
@@ -0,0 +1 @@
|
||||
Cloned from 991a76331cdf5d8f9dbf5b18f6e29adc80749a2f https://github.com/greenbender/pynntp/commit/991a76331cdf5d8f9dbf5b18f6e29adc80749a2f
|
||||
@@ -0,0 +1,59 @@
|
||||
pynntp
|
||||
======
|
||||
|
||||
Python NNTP library.
|
||||
|
||||
This package includes advanced NNTP features, including, compressed headers.
|
||||
|
||||
The most important (useful) feature of this package over other nntp libaries is
|
||||
the ablity to use generators to produce data. This allows for streaming download
|
||||
of large responses to say an XOVER command (which can produce gigabytes of data)
|
||||
and allows you to process the data at the same time is is being received.
|
||||
Meaning that memory use is minimal (even for the largest responses) and that
|
||||
cycles aren't being wasted waiting on a blocking read (even in a single threaded
|
||||
application)
|
||||
|
||||
|
||||
Example
|
||||
-------
|
||||
|
||||
>>> import nntp
|
||||
>>> nntp_client = nntp.NNTPClient("usenet-host.com", 443, "user", "password", use_ssl=True)
|
||||
>>> nntp_client.date()
|
||||
datetime.datetime(2013, 10, 19, 6, 11, 41, tzinfo=_tzgmt())
|
||||
>>> nntp_client.xfeature_compress_gzip()
|
||||
True
|
||||
>>> nntp_client.date()
|
||||
datetime.datetime(2013, 10, 19, 6, 13, 3, tzinfo=_tzgmt())
|
||||
|
||||
|
||||
Supported Commands
|
||||
------------------
|
||||
|
||||
NNTP commands that are currently supported include:
|
||||
* CAPABILITIES
|
||||
* MODE READER
|
||||
* QUIT
|
||||
* DATE
|
||||
* HELP
|
||||
* NEWGROUPS (generator)
|
||||
* NEWNEWS (generator)
|
||||
* LIST ACTIVE (generator)
|
||||
* LIST ACTIVE.TIMES (generator)
|
||||
* LIST NEWSGROUPS (generator)
|
||||
* LIST OVERVIEW.FMT (generator)
|
||||
* LIST EXTENSIONS (generator)
|
||||
* GROUP
|
||||
* NEXT
|
||||
* LAST
|
||||
* ARTICLE
|
||||
* HEAD
|
||||
* BODY
|
||||
* POST
|
||||
* XGTITLE
|
||||
* XHDR
|
||||
* XZHDR
|
||||
* XOVER (generator)
|
||||
* XZVER (generator)
|
||||
* XPAT (generator)
|
||||
* XFEATURE COMPRESS GZIP
|
||||
@@ -0,0 +1,10 @@
|
||||
from nntp import (
|
||||
NNTPError,
|
||||
NNTPReplyError,
|
||||
NNTPTemporaryError,
|
||||
NNTPPermanentError,
|
||||
NNTPProtocolError,
|
||||
NNTPDataError,
|
||||
BaseNNTPClient,
|
||||
NNTPClient,
|
||||
)
|
||||
@@ -0,0 +1,527 @@
|
||||
#!/usr/bin/python
|
||||
"""
|
||||
Date utilities to do fast datetime parsing.
|
||||
Copyright (C) 2013 Byron Platt
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
"""
|
||||
|
||||
# TODO: At the moment this has been targeted toward the datetime formats used by
|
||||
# NNTP as it was developed for use in a NNTP reader. There is, however, no
|
||||
# reason why this module could not be extended to include other formats.
|
||||
|
||||
import calendar
|
||||
import datetime
|
||||
import dateutil.parser
|
||||
import dateutil.tz
|
||||
|
||||
class _tzgmt(dateutil.tz.tzutc):
|
||||
"""GMT timezone.
|
||||
"""
|
||||
|
||||
def tzname(self, dt):
|
||||
return "GMT"
|
||||
|
||||
TZ_LOCAL = dateutil.tz.tzlocal()
|
||||
"""Local timezone (at the time the module was loaded)"""
|
||||
|
||||
TZ_UTC = dateutil.tz.tzutc()
|
||||
"""UTC timezone."""
|
||||
|
||||
TZ_GMT = _tzgmt()
|
||||
"""GMT timezone."""
|
||||
|
||||
_months = dict(
|
||||
jan=1, feb=2, mar=3, apr=4, may=5, jun=6,
|
||||
jul=7, aug=8, sep=9, oct=10,nov=11,dec=12
|
||||
)
|
||||
"""Conversion dictionary for english abbreviated month to integer."""
|
||||
|
||||
def _offset(value):
|
||||
"""Parse timezone to offset in seconds.
|
||||
|
||||
Args:
|
||||
value: A timezone in the '+0000' format. An integer would also work.
|
||||
|
||||
Returns:
|
||||
The timezone offset from GMT in seconds as an integer.
|
||||
"""
|
||||
o = int(value)
|
||||
if o == 0:
|
||||
return 0
|
||||
a = abs(o)
|
||||
s = a*36+(a%100)*24
|
||||
return (o//a)*s
|
||||
|
||||
def timestamp_d_b_Y_H_M_S(value):
|
||||
"""Convert timestamp string to time in seconds since epoch.
|
||||
|
||||
Timestamps strings like '18 Jun 2013 12:00:00 GMT' are able to be converted
|
||||
by this function.
|
||||
|
||||
Args:
|
||||
value: A timestamp string in the format '%d %b %Y %H:%M:%S GMT'.
|
||||
|
||||
Returns:
|
||||
The time in seconds since epoch as an integer.
|
||||
|
||||
Raises:
|
||||
ValueError: If timestamp is invalid.
|
||||
KeyError: If the abbrieviated month is invalid.
|
||||
|
||||
Note: The timezone is ignored it is simply assumed to be UTC/GMT.
|
||||
"""
|
||||
d, b, Y, t, Z = value.split()
|
||||
H, M, S = t.split(":")
|
||||
return int(calendar.timegm((
|
||||
int(Y), _months[b.lower()], int(d), int(H), int(M), int(S), 0, 0, 0
|
||||
)))
|
||||
|
||||
def datetimeobj_d_b_Y_H_M_S(value):
|
||||
"""Convert timestamp string to a datetime object.
|
||||
|
||||
Timestamps strings like '18 Jun 2013 12:00:00 GMT' are able to be converted
|
||||
by this function.
|
||||
|
||||
Args:
|
||||
value: A timestamp string in the format '%d %b %Y %H:%M:%S GMT'.
|
||||
|
||||
Returns:
|
||||
A datetime object.
|
||||
|
||||
Raises:
|
||||
ValueError: If timestamp is invalid.
|
||||
KeyError: If the abbrieviated month is invalid.
|
||||
|
||||
Note: The timezone is ignored it is simply assumed to be UTC/GMT.
|
||||
"""
|
||||
d, b, Y, t, Z = value.split()
|
||||
H, M, S = t.split(":")
|
||||
return datetime.datetime(
|
||||
int(Y), _months[b.lower()], int(d), int(H), int(M), int(S), tzinfo=TZ_GMT
|
||||
)
|
||||
|
||||
def timestamp_a__d_b_Y_H_M_S_z(value):
|
||||
"""Convert timestamp string to time in seconds since epoch.
|
||||
|
||||
Timestamps strings like 'Tue, 18 Jun 2013 22:00:00 +1000' are able to be
|
||||
converted by this function.
|
||||
|
||||
Args:
|
||||
value: A timestamp string in the format '%a, %d %b %Y %H:%M:%S %z'.
|
||||
|
||||
Returns:
|
||||
The time in seconds since epoch as an integer.
|
||||
|
||||
Raises:
|
||||
ValueError: If timestamp is invalid.
|
||||
KeyError: If the abbrieviated month is invalid.
|
||||
"""
|
||||
a, d, b, Y, t, z = value.split()
|
||||
H, M, S = t.split(":")
|
||||
return int(calendar.timegm((
|
||||
int(Y), _months[b.lower()], int(d), int(H), int(M), int(S), 0, 0, 0
|
||||
))) - _offset(z)
|
||||
|
||||
def datetimeobj_a__d_b_Y_H_M_S_z(value):
|
||||
"""Convert timestamp string to a datetime object.
|
||||
|
||||
Timestamps strings like 'Tue, 18 Jun 2013 22:00:00 +1000' are able to be
|
||||
converted by this function.
|
||||
|
||||
Args:
|
||||
value: A timestamp string in the format '%a, %d %b %Y %H:%M:%S %z'.
|
||||
|
||||
Returns:
|
||||
A datetime object.
|
||||
|
||||
Raises:
|
||||
ValueError: If timestamp is invalid.
|
||||
KeyError: If the abbrieviated month is invalid.
|
||||
"""
|
||||
a, d, b, Y, t, z = value.split()
|
||||
H, M, S = t.split(":")
|
||||
return datetime.datetime(
|
||||
int(Y), _months[b.lower()], int(d), int(H), int(M), int(S),
|
||||
tzinfo=dateutil.tz.tzoffset(None, _offset(z))
|
||||
)
|
||||
|
||||
def timestamp_YmdHMS(value):
|
||||
"""Convert timestamp string to time in seconds since epoch.
|
||||
|
||||
Timestamps strings like '20130618120000' are able to be converted by this
|
||||
function.
|
||||
|
||||
Args:
|
||||
value: A timestamp string in the format '%Y%m%d%H%M%S'.
|
||||
|
||||
Returns:
|
||||
The time in seconds since epoch as an integer.
|
||||
|
||||
Raises:
|
||||
ValueError: If timestamp is invalid.
|
||||
|
||||
Note: The timezone is assumed to be UTC/GMT.
|
||||
"""
|
||||
i = int(value)
|
||||
S = i
|
||||
M = S//100
|
||||
H = M//100
|
||||
d = H//100
|
||||
m = d//100
|
||||
Y = m//100
|
||||
return int(calendar.timegm((
|
||||
Y % 10000, m % 100, d % 100, H % 100, M % 100, S % 100, 0, 0, 0)
|
||||
))
|
||||
|
||||
def datetimeobj_YmdHMS(value):
|
||||
"""Convert timestamp string to a datetime object.
|
||||
|
||||
Timestamps strings like '20130618120000' are able to be converted by this
|
||||
function.
|
||||
|
||||
Args:
|
||||
value: A timestamp string in the format '%Y%m%d%H%M%S'.
|
||||
|
||||
Returns:
|
||||
A datetime object.
|
||||
|
||||
Raises:
|
||||
ValueError: If timestamp is invalid.
|
||||
|
||||
Note: The timezone is assumed to be UTC/GMT.
|
||||
"""
|
||||
i = int(value)
|
||||
S = i
|
||||
M = S//100
|
||||
H = M//100
|
||||
d = H//100
|
||||
m = d//100
|
||||
Y = m//100
|
||||
return datetime.datetime(
|
||||
Y % 10000, m % 100, d % 100, H % 100, M % 100, S % 100, tzinfo=TZ_GMT
|
||||
)
|
||||
|
||||
def timestamp_epoch(value):
|
||||
"""Convert timestamp string to a datetime object.
|
||||
|
||||
Timestamps strings like '1383470155' are able to be converted by this
|
||||
function.
|
||||
|
||||
Args:
|
||||
value: A timestamp string as seconds since epoch.
|
||||
|
||||
Returns:
|
||||
The time in seconds since epoch as an integer.
|
||||
"""
|
||||
return int(value)
|
||||
|
||||
def datetimeobj_epoch(value):
|
||||
"""Convert timestamp string to a datetime object.
|
||||
|
||||
Timestamps strings like '1383470155' are able to be converted by this
|
||||
function.
|
||||
|
||||
Args:
|
||||
value: A timestamp string as seconds since epoch.
|
||||
|
||||
Returns:
|
||||
A datetime object.
|
||||
|
||||
Raises:
|
||||
ValueError: If timestamp is invalid.
|
||||
"""
|
||||
return datetime.datetime.utcfromtimestamp(int(value)).replace(tzinfo=TZ_GMT)
|
||||
|
||||
def timestamp_fmt(value, fmt):
|
||||
"""Convert timestamp string to time in seconds since epoch.
|
||||
|
||||
Wraps the datetime.datetime.strptime(). This is slow use the other
|
||||
timestamp_*() functions if possible.
|
||||
|
||||
Args:
|
||||
value: A timestamp string.
|
||||
fmt: A timestamp format string.
|
||||
|
||||
Returns:
|
||||
The time in seconds since epoch as an integer.
|
||||
"""
|
||||
return int(calendar.timegm(
|
||||
datetime.datetime.strptime(value, fmt).utctimetuple()
|
||||
))
|
||||
|
||||
def datetimeobj_fmt(value, fmt):
|
||||
"""Convert timestamp string to a datetime object.
|
||||
|
||||
Wrapper for datetime.datetime.strptime(). This is slow use the other
|
||||
timestamp_*() functions if possible.
|
||||
|
||||
Args:
|
||||
value: A timestamp string.
|
||||
fmt: A timestamp format string.
|
||||
|
||||
Returns:
|
||||
A datetime object.
|
||||
"""
|
||||
return datetime.datetime.strptime(value, fmt)
|
||||
|
||||
def timestamp_any(value):
|
||||
"""Convert timestamp string to time in seconds since epoch.
|
||||
|
||||
Most timestamps strings are supported in fact this wraps the
|
||||
dateutil.parser.parse() method. This is SLOW use the other timestamp_*()
|
||||
functions if possible.
|
||||
|
||||
Args:
|
||||
value: A timestamp string.
|
||||
|
||||
Returns:
|
||||
The time in seconds since epoch as an integer.
|
||||
"""
|
||||
return int(calendar.timegm(dateutil.parser.parse(value).utctimetuple()))
|
||||
|
||||
def datetimeobj_any(value):
|
||||
"""Convert timestamp string to a datetime object.
|
||||
|
||||
Most timestamps strings are supported in fact this is a wrapper for the
|
||||
dateutil.parser.parse() method. This is SLOW use the other datetimeobj_*()
|
||||
functions if possible.
|
||||
|
||||
Args:
|
||||
value: A timestamp string.
|
||||
|
||||
Returns:
|
||||
A datetime object.
|
||||
"""
|
||||
return dateutil.parser.parse(value)
|
||||
|
||||
_timestamp_formats = {
|
||||
"%d %b %Y %H:%M:%S" : timestamp_d_b_Y_H_M_S,
|
||||
"%a, %d %b %Y %H:%M:%S %z": timestamp_a__d_b_Y_H_M_S_z,
|
||||
"%Y%m%d%H%M%S" : timestamp_YmdHMS,
|
||||
"epoch" : timestamp_epoch,
|
||||
}
|
||||
|
||||
def timestamp(value, fmt=None):
|
||||
"""Parse a datetime to a unix timestamp.
|
||||
|
||||
Uses fast custom parsing for common datetime formats or the slow dateutil
|
||||
parser for other formats. This is a trade off between ease of use and speed
|
||||
and is very useful for fast parsing of timestamp strings whose format may
|
||||
standard but varied or unknown prior to parsing.
|
||||
|
||||
Common formats include:
|
||||
1 Feb 2010 12:00:00 GMT
|
||||
Mon, 1 Feb 2010 22:00:00 +1000
|
||||
20100201120000
|
||||
1383470155 (seconds since epoch)
|
||||
|
||||
See the other timestamp_*() functions for more details.
|
||||
|
||||
Args:
|
||||
value: A string representing a datetime.
|
||||
fmt: A timestamp format string like for time.strptime().
|
||||
|
||||
Returns:
|
||||
The time in seconds since epoch as and integer for the value specified.
|
||||
"""
|
||||
if fmt:
|
||||
return _timestamp_formats.get(fmt,
|
||||
lambda v: timestamp_fmt(v, fmt)
|
||||
)(value)
|
||||
|
||||
l = len(value)
|
||||
|
||||
if 19 <= l <= 24 and value[3] == " ":
|
||||
# '%d %b %Y %H:%M:%Sxxxx'
|
||||
try:
|
||||
return timestamp_d_b_Y_H_M_S(value)
|
||||
except (KeyError, ValueError, OverflowError):
|
||||
pass
|
||||
|
||||
if 30 <= l <= 31:
|
||||
# '%a, %d %b %Y %H:%M:%S %z'
|
||||
try:
|
||||
return timestamp_a__d_b_Y_H_M_S_z(value)
|
||||
except (KeyError, ValueError, OverflowError):
|
||||
pass
|
||||
|
||||
if l == 14:
|
||||
# '%Y%m%d%H%M%S'
|
||||
try:
|
||||
return timestamp_YmdHMS(value)
|
||||
except (ValueError, OverflowError):
|
||||
pass
|
||||
|
||||
# epoch timestamp
|
||||
try:
|
||||
return timestamp_epoch(value)
|
||||
except ValueError:
|
||||
pass
|
||||
|
||||
# slow version
|
||||
return timestamp_any(value)
|
||||
|
||||
_datetimeobj_formats = {
|
||||
"%d %b %Y %H:%M:%S" : datetimeobj_d_b_Y_H_M_S,
|
||||
"%a, %d %b %Y %H:%M:%S %z": datetimeobj_a__d_b_Y_H_M_S_z,
|
||||
"%Y%m%d%H%M%S" : datetimeobj_YmdHMS,
|
||||
"epoch" : datetimeobj_epoch,
|
||||
}
|
||||
|
||||
def datetimeobj(value, fmt=None):
|
||||
"""Parse a datetime to a datetime object.
|
||||
|
||||
Uses fast custom parsing for common datetime formats or the slow dateutil
|
||||
parser for other formats. This is a trade off between ease of use and speed
|
||||
and is very useful for fast parsing of timestamp strings whose format may
|
||||
standard but varied or unknown prior to parsing.
|
||||
|
||||
Common formats include:
|
||||
1 Feb 2010 12:00:00 GMT
|
||||
Mon, 1 Feb 2010 22:00:00 +1000
|
||||
20100201120000
|
||||
1383470155 (seconds since epoch)
|
||||
|
||||
See the other datetimeobj_*() functions for more details.
|
||||
|
||||
Args:
|
||||
value: A string representing a datetime.
|
||||
|
||||
Returns:
|
||||
A datetime object.
|
||||
"""
|
||||
if fmt:
|
||||
return _datetimeobj_formats.get(fmt,
|
||||
lambda v: datetimeobj_fmt(v, fmt)
|
||||
)(value)
|
||||
|
||||
l = len(value)
|
||||
|
||||
if 19 <= l <= 24 and value[3] == " ":
|
||||
# '%d %b %Y %H:%M:%Sxxxx'
|
||||
try:
|
||||
return datetimeobj_d_b_Y_H_M_S(value)
|
||||
except (KeyError, ValueError):
|
||||
pass
|
||||
|
||||
if 30 <= l <= 31:
|
||||
# '%a, %d %b %Y %H:%M:%S %z'
|
||||
try:
|
||||
return datetimeobj_a__d_b_Y_H_M_S_z(value)
|
||||
except (KeyError, ValueError):
|
||||
pass
|
||||
|
||||
if l == 14:
|
||||
# '%Y%m%d%H%M%S'
|
||||
try:
|
||||
return datetimeobj_YmdHMS(value)
|
||||
except ValueError:
|
||||
pass
|
||||
|
||||
# epoch timestamp
|
||||
try:
|
||||
return datetimeobj_epoch(value)
|
||||
except ValueError:
|
||||
pass
|
||||
|
||||
# slow version
|
||||
return datetimeobj_any(value)
|
||||
|
||||
|
||||
# testing
|
||||
if __name__ == "__main__":
|
||||
|
||||
import sys
|
||||
import timeit
|
||||
|
||||
log = sys.stdout.write
|
||||
|
||||
times = (
|
||||
datetime.datetime.now(TZ_UTC),
|
||||
datetime.datetime.now(TZ_GMT),
|
||||
datetime.datetime.now(TZ_LOCAL),
|
||||
datetime.datetime.now(),
|
||||
)
|
||||
|
||||
# check timezones
|
||||
for t in times:
|
||||
log("%s\n" % t.strftime("%Y-%m-%d %H:%M:%S %Z"))
|
||||
|
||||
# TODO validate values (properly)
|
||||
|
||||
# check speed
|
||||
values = (
|
||||
{
|
||||
"name": "Implemented Format",
|
||||
"time": "20130624201912",
|
||||
"fmt" : "%Y%m%d%H%M%S"
|
||||
},
|
||||
{
|
||||
"name": "Unimplemented Format",
|
||||
"time": "2013-06-24 20:19:12",
|
||||
"fmt" : "%Y-%m-%d %H:%M:%S"
|
||||
}
|
||||
)
|
||||
tests = (
|
||||
{
|
||||
"name" : "GMT timestamp (strptime version)",
|
||||
"test" : "int(calendar.timegm(datetime.datetime.strptime('%(time)s', '%(fmt)s').utctimetuple()))",
|
||||
"setup": "import calendar, datetime",
|
||||
},
|
||||
{
|
||||
"name" : "GMT timestamp (dateutil version)",
|
||||
"test" : "int(calendar.timegm(dateutil.parser.parse('%(time)s').utctimetuple()))",
|
||||
"setup": "import calendar, dateutil.parser",
|
||||
},
|
||||
{
|
||||
"name" : "GMT timestamp (fast version)",
|
||||
"test" : "timestamp('%(time)s')",
|
||||
"setup": "from __main__ import timestamp",
|
||||
},
|
||||
{
|
||||
"name" : "GMT timestamp (fast version with format hint)",
|
||||
"test" : "timestamp('%(time)s', '%(fmt)s')",
|
||||
"setup": "from __main__ import timestamp",
|
||||
},
|
||||
{
|
||||
"name" : "GMT datetime object (strptime version)",
|
||||
"test" : "datetime.datetime.strptime('%(time)s', '%(fmt)s').replace(tzinfo=TZ_GMT)",
|
||||
"setup": "import datetime; from __main__ import TZ_GMT",
|
||||
},
|
||||
{
|
||||
"name" : "GMT datetime object (dateutil version)",
|
||||
"test" : "dateutil.parser.parse('%(time)s').replace(tzinfo=TZ_GMT)",
|
||||
"setup": "import dateutil.parser; from __main__ import TZ_GMT",
|
||||
},
|
||||
{
|
||||
"name" : "GMT datetime object (fast version)",
|
||||
"test" : "datetimeobj('%(time)s')",
|
||||
"setup": "from __main__ import datetimeobj",
|
||||
},
|
||||
{
|
||||
"name" : "GMT datetime object (fast version with format hint)",
|
||||
"test" : "datetimeobj('%(time)s', '%(fmt)s')",
|
||||
"setup": "from __main__ import datetimeobj",
|
||||
}
|
||||
)
|
||||
iters = 100000
|
||||
for v in values:
|
||||
log("%(name)s (%(fmt)s)\n" % v)
|
||||
for t in tests:
|
||||
log(" %(name)-52s" % t)
|
||||
elapsed = timeit.timeit(t["test"] % v, t["setup"], number=iters)
|
||||
log("%0.3f sec (%d loops @ %0.3f usec)\n" % (
|
||||
elapsed, iters, (elapsed/iters)*1000000
|
||||
))
|
||||
@@ -0,0 +1,150 @@
|
||||
#!/usr/bin/python
|
||||
"""
|
||||
A reasonably efficient FIFO buffer.
|
||||
Copyright (C) 2013 Byron Platt
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
"""
|
||||
|
||||
#NOTE: docstrings beed to be added
|
||||
|
||||
_DISCARD_SIZE = 0xffff
|
||||
|
||||
class Fifo(object):
|
||||
|
||||
def __init__(self, data="", eol="\r\n"):
|
||||
|
||||
self.buf = data
|
||||
self.eol = eol
|
||||
self.buflist = []
|
||||
self.pos = 0
|
||||
|
||||
def __len__(self):
|
||||
|
||||
return len(self.buf) - self.pos
|
||||
|
||||
def __iter__(self):
|
||||
|
||||
return self
|
||||
|
||||
def __discard(self):
|
||||
|
||||
if self.pos > _DISCARD_SIZE:
|
||||
self.buf = self.buf[self.pos:]
|
||||
self.pos = 0
|
||||
|
||||
def __append(self):
|
||||
|
||||
self.buf += "".join(self.buflist)
|
||||
self.buflist = []
|
||||
|
||||
def clear(self):
|
||||
|
||||
self.buf = ""
|
||||
self.buflist = []
|
||||
self.pos = 0
|
||||
|
||||
def write(self, data):
|
||||
|
||||
self.buflist.append(data)
|
||||
|
||||
def read(self, length=0):
|
||||
|
||||
self.__append()
|
||||
|
||||
if 0 < length < len(self):
|
||||
newpos = self.pos + length
|
||||
data = self.buf[self.pos:newpos]
|
||||
self.pos = newpos
|
||||
self.__discard()
|
||||
return data
|
||||
|
||||
data = self.buf[self.pos:]
|
||||
self.clear()
|
||||
return data
|
||||
|
||||
def readline(self):
|
||||
|
||||
self.__append()
|
||||
|
||||
i = self.buf.find(self.eol, self.pos)
|
||||
if i < 0:
|
||||
return ""
|
||||
|
||||
newpos = i + len(self.eol)
|
||||
data = self.buf[self.pos:newpos]
|
||||
self.pos = newpos
|
||||
self.__discard()
|
||||
return data
|
||||
|
||||
def readuntil(self, token, size=0):
|
||||
|
||||
self.__append()
|
||||
|
||||
i = self.buf.find(token, self.pos)
|
||||
if i < 0:
|
||||
index = max(len(token) - 1, size)
|
||||
newpos = max(len(self.buf) - index, self.pos)
|
||||
data = self.buf[self.pos:newpos]
|
||||
self.pos = newpos
|
||||
self.__discard()
|
||||
return False, data
|
||||
|
||||
newpos = i + len(token)
|
||||
data = self.buf[self.pos:newpos]
|
||||
self.pos = newpos
|
||||
self.__discard()
|
||||
return True, data
|
||||
|
||||
def peek(self, length=0):
|
||||
|
||||
self.__append()
|
||||
|
||||
if 0 < length < len(self):
|
||||
newpos = self.pos + length
|
||||
return self.buf[self.pos:newpos]
|
||||
|
||||
return self.buf[self.pos:]
|
||||
|
||||
def peekline(self):
|
||||
|
||||
self.__append()
|
||||
|
||||
i = self.buf.find(self.eol, self.pos)
|
||||
if i < 0:
|
||||
return ""
|
||||
|
||||
newpos = i + len(self.eol)
|
||||
return self.buf[self.pos:newpos]
|
||||
|
||||
def peekuntil(self, token, size=0):
|
||||
|
||||
self.__append()
|
||||
|
||||
i = self.buf.find(token, self.pos)
|
||||
if i < 0:
|
||||
index = max(len(token) - 1, size)
|
||||
newpos = max(len(self.buf) - index, self.pos)
|
||||
return False, self.buf[self.pos:newpos]
|
||||
|
||||
newpos = i + len(token)
|
||||
return True, self.buf[self.pos:newpos]
|
||||
|
||||
def next(self):
|
||||
|
||||
line = self.readline()
|
||||
if not line:
|
||||
raise StopIteration()
|
||||
|
||||
return line
|
||||
@@ -0,0 +1,103 @@
|
||||
#!/usr/bin/python
|
||||
"""
|
||||
Case-insentitive ordered dictionary (useful for headers).
|
||||
Copyright (C) 2013 Byron Platt
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
"""
|
||||
|
||||
#NOTE: docstrings beed to be added
|
||||
|
||||
from collections import OrderedDict
|
||||
|
||||
def _lower(v):
|
||||
"""assumes that classes that inherit list, tuple or dict have a constructor
|
||||
that is compatible with those base classes. If you are using classes that
|
||||
don't satisfy this requirement you can subclass them and add a lower()
|
||||
method for the class"""
|
||||
if hasattr(v, "lower"):
|
||||
return v.lower()
|
||||
if isinstance(v, (list, tuple)):
|
||||
return v.__class__(_lower(x) for x in v)
|
||||
if isinstance(v, dict):
|
||||
return v.__class__(_lower(v.items()))
|
||||
return v
|
||||
|
||||
|
||||
# NOTE: This class makes assumptions about the OrderedDict class that it
|
||||
# inherits from -- this a a bad idea. If The OrderedDict Class were to change
|
||||
# behaviour it could break this class.
|
||||
class IODict(OrderedDict):
|
||||
"""Case in-sensitive ordered dictionary.
|
||||
>>> iod = IODict([('ABC', 1), ('DeF', 'A'), (('gHi', 'jkl', 20), 'b')])
|
||||
>>> iod
|
||||
IODict([('ABC', 1), ('DeF', 'A'), (('gHi', 'jkl', 20), 'b')])
|
||||
>>> iod['ABC'], iod['abc'], iod['aBc']
|
||||
(1, 1, 1)
|
||||
>>> iod['DeF'], iod['def'], iod['dEf']
|
||||
('A', 'A', 'A')
|
||||
>>> iod[('gHi', 'jkl', 20)], iod[('ghi', 'jKL', 20)]
|
||||
('b', 'b')
|
||||
>>> iod == {"aBc": 1, "deF": 'A', ('Ghi', 'JKL', 20): 'b'}
|
||||
True
|
||||
>>> iod.popitem()
|
||||
(('gHi', 'jkl', 20), 'b')
|
||||
"""
|
||||
|
||||
def __init__(self, *args, **kwds):
|
||||
self.__map = {}
|
||||
OrderedDict.__init__(self, *args, **kwds)
|
||||
|
||||
def __setitem__(self, key, *args, **kwds):
|
||||
l = _lower(key)
|
||||
OrderedDict.__setitem__(self, l, *args, **kwds)
|
||||
self.__map[l] = key
|
||||
|
||||
def __getitem__(self, key, *args, **kwds):
|
||||
l = _lower(key)
|
||||
return OrderedDict.__getitem__(self, l, *args, **kwds)
|
||||
|
||||
def __delitem__(self, key, *args, **kwds):
|
||||
l = _lower(key)
|
||||
OrderedDict.__delitem__(self, l, *args, **kwds)
|
||||
del self.__map[l]
|
||||
|
||||
def __contains__(self, key):
|
||||
l = _lower(key)
|
||||
return OrderedDict.__contains__(self, l)
|
||||
|
||||
def __iter__(self):
|
||||
for k in OrderedDict.__iter__(self):
|
||||
yield self.__map[k]
|
||||
|
||||
def __reversed__(self):
|
||||
for k in OrderedDict.__reversed__(self):
|
||||
yield self.__map[k]
|
||||
|
||||
def clear(self):
|
||||
OrderedDict.clear(self)
|
||||
self.__map.clear()
|
||||
|
||||
def __eq__(self, other):
|
||||
"""assumes that classes that inherit dict have a constructor that is
|
||||
compatible with the dict class."""
|
||||
if len(self) != len(other) or not isinstance(other, dict):
|
||||
return False
|
||||
so = OrderedDict(zip(_lower(self.keys()), self.values()))
|
||||
oo = other.__class__(zip(_lower(other.keys()), other.values()))
|
||||
return so == oo
|
||||
|
||||
if __name__ == "__main__":
|
||||
import doctest
|
||||
doctest.testmod()
|
||||
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,229 @@
|
||||
#!/usr/bin/python
|
||||
"""
|
||||
An NNTP library - a bit more useful than the nntplib one (hopefully).
|
||||
Copyright (C) 2013 Byron Platt
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
"""
|
||||
|
||||
import cStringIO
|
||||
import iodict
|
||||
|
||||
def unparse_msgid_article(obj):
|
||||
"""Unparse a message-id or article number argument.
|
||||
|
||||
Args:
|
||||
obj: A messsage id or an article number.
|
||||
|
||||
Returns:
|
||||
The message id or article number as a string.
|
||||
"""
|
||||
return str(obj)
|
||||
|
||||
def parse_msgid_article(obj):
|
||||
"""Parse a message-id or article number argument.
|
||||
|
||||
Args:
|
||||
str: Message id or article as a string.
|
||||
|
||||
Returns:
|
||||
The message id as a string or article number as an integer.
|
||||
"""
|
||||
try:
|
||||
return int(obj)
|
||||
except ValueError:
|
||||
pass
|
||||
return obj
|
||||
|
||||
def unparse_range(obj):
|
||||
"""Unparse a range argument.
|
||||
|
||||
Args:
|
||||
obj: An article range. There are a number of valid formats; an integer
|
||||
specifying a single article or a tuple specifying an article range.
|
||||
If the range doesn't give a start article then all articles up to
|
||||
the specified last article are included. If the range doesn't
|
||||
specify a last article then all articles from the first specified
|
||||
article up to the current last article for the group are included.
|
||||
|
||||
Returns:
|
||||
The range as a string that can be used by an NNTP command.
|
||||
|
||||
Note: Sample valid formats.
|
||||
4678
|
||||
(,5234)
|
||||
(4245,)
|
||||
(4245, 5234)
|
||||
"""
|
||||
if isinstance(obj, (int, long)):
|
||||
return str(obj)
|
||||
|
||||
if isinstance(obj, tuple):
|
||||
arg = str(obj[0]) + "-"
|
||||
if len(obj) > 1:
|
||||
arg += str(obj[1])
|
||||
return arg
|
||||
|
||||
raise ValueError("Must be an integer or tuple")
|
||||
|
||||
def parse_range(obj):
|
||||
"""Parse a range argument.
|
||||
|
||||
Args:
|
||||
obj: An article range as a string.
|
||||
|
||||
Returns:
|
||||
The range as a string that can be used by an NNTP command.
|
||||
|
||||
Raises:
|
||||
ValueError: If obj is not a valid range format.
|
||||
|
||||
Note: Sample valid formats.
|
||||
4678
|
||||
(,5234)
|
||||
(4245,)
|
||||
(4245, 5234)
|
||||
"""
|
||||
if isinstance(obj, (int, long)):
|
||||
return str(obj)
|
||||
|
||||
if isinstance(obj, tuple):
|
||||
arg = str(obj[0]) + "-"
|
||||
if len(obj) > 1:
|
||||
arg += str(obj[1])
|
||||
return arg
|
||||
|
||||
raise ValueError("Must be an integer or tuple")
|
||||
|
||||
def unparse_msgid_range(obj):
|
||||
"""Unparse a message-id or range argument.
|
||||
|
||||
Args:
|
||||
obj: A message id as a string or a range as specified by
|
||||
unparse_range().
|
||||
|
||||
Raises:
|
||||
ValueError: If obj is not a valid message id or range format. See
|
||||
unparse_range() for valid range formats.
|
||||
|
||||
Returns:
|
||||
A message id or range as a string that can be used by an NNTP command.
|
||||
"""
|
||||
if isinstance(obj, basestring):
|
||||
return obj
|
||||
|
||||
return unparse_range(obj)
|
||||
|
||||
def parse_newsgroup(line):
|
||||
"""Parse a newsgroup info line to python types.
|
||||
|
||||
Args:
|
||||
line: An info response line containing newsgroup info.
|
||||
|
||||
Returns:
|
||||
A tuple of group name, low-water as integer, high-water as integer and
|
||||
posting status.
|
||||
|
||||
Raises:
|
||||
ValueError: If the newsgroup info cannot be parsed.
|
||||
|
||||
Note:
|
||||
Posting status is a character is one of (but not limited to):
|
||||
"y" posting allowed
|
||||
"n" posting not allowed
|
||||
"m" posting is moderated
|
||||
"""
|
||||
parts = line.split()
|
||||
try:
|
||||
group = parts[0]
|
||||
low = int(parts[1])
|
||||
high = int(parts[2])
|
||||
status = parts[3]
|
||||
except (IndexError, ValueError):
|
||||
raise ValueError("Invalid newsgroup info")
|
||||
return group, low, high, status
|
||||
|
||||
def parse_header(line):
|
||||
"""Parse a header line.
|
||||
|
||||
Args:
|
||||
line: A header line as a string.
|
||||
|
||||
Returns:
|
||||
None if end of headers is found. A string giving the continuation line
|
||||
if a continuation is found. A tuple of name, value when a header line is
|
||||
found.
|
||||
|
||||
Raises:
|
||||
ValueError: If the line cannot be parsed as a header.
|
||||
"""
|
||||
if not line or line == "\r\n":
|
||||
return None
|
||||
if line[0] in " \t":
|
||||
return line[1:].rstrip()
|
||||
name, value = line.split(":", 1)
|
||||
return (name.strip(), value.strip())
|
||||
|
||||
def parse_headers(obj):
|
||||
"""Parse a string a iterable object (including file like objects) to a
|
||||
python dictionary.
|
||||
|
||||
Args:
|
||||
obj: An iterable object including file-like objects.
|
||||
|
||||
Returns:
|
||||
An dictionary of headers. If a header is repeated then the last value
|
||||
for that header is given.
|
||||
|
||||
Raises:
|
||||
ValueError: If the first line is a continuation line or the headers
|
||||
cannot be parsed.
|
||||
"""
|
||||
if isinstance(obj, basestring):
|
||||
obj = cStringIO.StringIO(obj)
|
||||
hdrs = []
|
||||
for line in obj:
|
||||
hdr = parse_header(line)
|
||||
if not hdr:
|
||||
break
|
||||
if isinstance(hdr, basestring):
|
||||
if not hdrs:
|
||||
raise ValueError("First header is a continuation")
|
||||
hdrs[-1] = (hdrs[-1][0], hdrs[-1][1] + hdr)
|
||||
continue
|
||||
hdrs.append(hdr)
|
||||
return iodict.IODict(hdrs)
|
||||
|
||||
def unparse_header(name, value):
|
||||
"""Parse a name value tuple to a header string.
|
||||
|
||||
Args:
|
||||
name: The header name.
|
||||
value: the header value.
|
||||
|
||||
Returns:
|
||||
The header as a string.
|
||||
"""
|
||||
return ": ".join([name, value]) + "\r\n"
|
||||
|
||||
def unparse_headers(hdrs):
|
||||
"""Parse a dictionary of headers to a string.
|
||||
|
||||
Args:
|
||||
hdrs: A dictionary of headers.
|
||||
|
||||
Returns:
|
||||
The headers as a string that can be used in an NNTP POST.
|
||||
"""
|
||||
return "".join([unparse_header(n, v) for n, v in hdrs.items()]) + "\r\n"
|
||||
@@ -0,0 +1,61 @@
|
||||
#!/usr/bin/python
|
||||
"""
|
||||
Basic yEnc decoder.
|
||||
Copyright (C) 2013 Byron Platt
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
"""
|
||||
|
||||
#NOTE: docstrings beed to be added
|
||||
|
||||
import re
|
||||
import zlib
|
||||
import struct
|
||||
import binascii
|
||||
|
||||
|
||||
_crc32_re = re.compile(r"\s+crc(?:32)?=([0-9a-fA-F]{8})")
|
||||
|
||||
def crc32(trailer):
|
||||
|
||||
match = _crc32_re.search(trailer)
|
||||
if not match:
|
||||
return None
|
||||
|
||||
buf = binascii.unhexlify(match.group(1))
|
||||
|
||||
return struct.unpack(">I", buf)[0]
|
||||
|
||||
|
||||
def decode(buff, escape=0, crc32=0):
|
||||
|
||||
decoded = ""
|
||||
|
||||
for c in buff:
|
||||
byte = ord(c)
|
||||
if escape:
|
||||
byte = (byte - 106) & 0xff
|
||||
escape = 0
|
||||
elif byte == 0x3d:
|
||||
escape = 1
|
||||
continue
|
||||
elif byte == 0x0d or byte == 0x0a:
|
||||
continue
|
||||
else:
|
||||
byte = (byte - 42) & 0xff
|
||||
decoded += chr(byte)
|
||||
|
||||
crc32 = zlib.crc32(decoded, crc32)
|
||||
|
||||
return decoded, escape, crc32
|
||||
@@ -0,0 +1,13 @@
|
||||
from setuptools import setup
|
||||
|
||||
setup(
|
||||
name='pynntp',
|
||||
version='0.8.4',
|
||||
description='NNTP Library (including compressed headers)',
|
||||
author='Byron Platt',
|
||||
author_email='byron.platt@gmail.com',
|
||||
license='GPL3',
|
||||
url='https://github.com/greenbender/pynntp',
|
||||
packages=['nntp'],
|
||||
install_requires=['dateutils'],
|
||||
)
|
||||
@@ -0,0 +1,246 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import json
|
||||
import time
|
||||
import random
|
||||
import socket
|
||||
import SocketServer
|
||||
import socketpool
|
||||
from lib.pynntp.nntp import nntp
|
||||
from lib.info import bcolors
|
||||
from socketpool import util
|
||||
|
||||
|
||||
class NNTPClientConnector(socketpool.Connector, nntp.NNTPClient):
|
||||
|
||||
def __init__(self, host, port, backend_mod, pool=None,
|
||||
username="anonymous", password="anonymous", timeout=60, use_ssl=False):
|
||||
self.host = host
|
||||
self.port = port
|
||||
self.backend_mod = backend_mod
|
||||
self._pool = pool
|
||||
self._connected = False
|
||||
self._life = time.time() - random.randint(0, 10)
|
||||
self._start_time = time.time()
|
||||
if backend_mod.Socket != socket.socket:
|
||||
raise ValueError("Bad backend")
|
||||
nntp.NNTPClient.__init__(self, self.host, self.port, username, password, timeout=timeout, use_ssl=use_ssl)
|
||||
self.id = self.socket.getsockname()[1]
|
||||
print(bcolors.PRIMARY + "New NNTP connection to %s established with id #%5d" %
|
||||
(self.host, self.id) + bcolors.ENDC)
|
||||
self._connected = True
|
||||
self.xfeature_compress_gzip()
|
||||
|
||||
def __del__(self):
|
||||
self.release()
|
||||
|
||||
def matches(self, **match_options):
|
||||
target_host = match_options.get('host')
|
||||
target_port = match_options.get('port')
|
||||
return target_host == self.host and target_port == self.port
|
||||
|
||||
def is_connected(self):
|
||||
if self._connected:
|
||||
return socketpool.util.is_connected(self.socket)
|
||||
return False
|
||||
|
||||
def handle_exception(self, exception):
|
||||
print(bcolors.ERROR + str(exception) + bcolors.ENDC)
|
||||
self.release()
|
||||
self.invalidate()
|
||||
|
||||
def get_lifetime(self):
|
||||
return self._life
|
||||
|
||||
def invalidate(self):
|
||||
print(bcolors.PRIMARY + "Disconnecting from NNTP connection id #%5d after %d seconds." %
|
||||
(self.id, (time.time() - self._start_time)) + bcolors.ENDC)
|
||||
self.close()
|
||||
self._connected = False
|
||||
self._life = -1
|
||||
|
||||
def release(self):
|
||||
if self._pool is not None:
|
||||
if self._connected:
|
||||
self._pool.release_connection(self)
|
||||
else:
|
||||
self._pool = None
|
||||
|
||||
|
||||
# NNTP proxy request handler for newznab
|
||||
class NNTPProxyRequestHandler(SocketServer.StreamRequestHandler):
|
||||
|
||||
def handle(self):
|
||||
with self.server.nntp_client_pool.connection() as nntp_client:
|
||||
|
||||
self.wfile.write("200 localhost NNRP Service Ready.\r\n")
|
||||
|
||||
for line in self.rfile:
|
||||
data = line.strip()
|
||||
|
||||
if not data.startswith("POST"):
|
||||
if data.startswith("GROUP"):
|
||||
print(bcolors.ALTERNATE + "%5d %s" % (nntp_client.id, data))
|
||||
else:
|
||||
print(bcolors.HEADER + "%5d " % nntp_client.id) + (bcolors.PRIMARY + "%s" % data)
|
||||
|
||||
if data.startswith("XOVER"):
|
||||
try:
|
||||
rng = data.split(None, 1)[1]
|
||||
rng = tuple(map(int, rng.split("-")))
|
||||
xover_gen = nntp_client.xover_gen(rng)
|
||||
self.wfile.write("224 data follows\r\n")
|
||||
for entry in xover_gen:
|
||||
self.wfile.write("\t".join(entry) + "\r\n")
|
||||
self.wfile.write(".\r\n")
|
||||
except Exception as ex:
|
||||
print(bcolors.ERROR + str(ex.message) + bcolors.ENDC)
|
||||
self.wfile.write("503 internal server error\r\n")
|
||||
|
||||
elif data.startswith("BODY"):
|
||||
msgid = data.split(None, 1)[1]
|
||||
try:
|
||||
body = nntp_client.body(msgid)
|
||||
self.wfile.write("222 %s\r\n" % msgid)
|
||||
self.wfile.write(body.replace("\r\n.", "\r\n.."))
|
||||
self.wfile.write(".\r\n")
|
||||
except Exception as ex:
|
||||
print(bcolors.ERROR + str(ex.message) + bcolors.ENDC)
|
||||
self.wfile.write("430 no such article\r\n")
|
||||
|
||||
elif data.startswith("GROUP"):
|
||||
try:
|
||||
total, first, last, group = nntp_client.group(data.split(None, 1)[1])
|
||||
self.wfile.write("211 %d %d %d %s\r\n" % (total, first, last, group))
|
||||
except Exception as ex:
|
||||
print(bcolors.ERROR + str(ex.message) + bcolors.ENDC)
|
||||
self.wfile.write("411 no such news group\r\n")
|
||||
|
||||
elif data.startswith("LIST OVERVIEW.FMT"):
|
||||
try:
|
||||
fmt = nntp_client.list_overview_fmt()
|
||||
self.wfile.write("215 Order of fields in overview database.\r\n")
|
||||
fmt = "\r\n".join(["%s:%s" % (f[0], "full" if f[1] else "") for f in fmt]) + "\r\n"
|
||||
self.wfile.write(fmt)
|
||||
self.wfile.write(".\r\n")
|
||||
except Exception as ex:
|
||||
print(bcolors.ERROR + str(ex.message) + bcolors.ENDC)
|
||||
self.wfile.write("503 internal server error\r\n")
|
||||
|
||||
elif data.startswith("HEAD"):
|
||||
msgid = data.split(None, 1)[1]
|
||||
try:
|
||||
head = nntp_client.head(msgid)
|
||||
self.wfile.write("221 %s\r\n" % msgid)
|
||||
head = "\r\n".join([": ".join(item) for item in head.items()]) + "\r\n\r\n"
|
||||
self.wfile.write(head)
|
||||
self.wfile.write(".\r\n")
|
||||
except Exception as ex:
|
||||
print(bcolors.ERROR + str(ex.message) + bcolors.ENDC)
|
||||
self.wfile.write("430 no such article\r\n")
|
||||
|
||||
elif data.startswith("ARTICLE"):
|
||||
msgid = data.split(None, 1)[1]
|
||||
try:
|
||||
article = nntp_client.article(msgid, False)
|
||||
# check no of return values for compatibility with pynntp<=0.8.3
|
||||
if len(article) == 2:
|
||||
articleno, head, body = 0, article[0], article[1]
|
||||
else:
|
||||
articleno, head, body = article
|
||||
self.wfile.write("220 %d %s\r\n" % (articleno, msgid))
|
||||
head = "\r\n".join([": ".join(item) for item in head.items()]) + "\r\n\r\n"
|
||||
self.wfile.write(head)
|
||||
self.wfile.write(body.replace("\r\n.", "\r\n.."))
|
||||
self.wfile.write(".\r\n")
|
||||
except Exception as ex:
|
||||
print(bcolors.ERROR + str(ex.message) + bcolors.ENDC)
|
||||
self.wfile.write("430 no such article\r\n")
|
||||
|
||||
elif data == "LIST":
|
||||
try:
|
||||
list_gen = nntp_client.list_gen()
|
||||
self.wfile.write("215 list of newsgroups follows\r\n")
|
||||
for entry in list_gen:
|
||||
self.wfile.write("%s %d %d %s\r\n" % entry)
|
||||
self.wfile.write(".\r\n")
|
||||
except Exception as ex:
|
||||
print(bcolors.ERROR + str(ex.message) + bcolors.ENDC)
|
||||
self.wfile.write("503 internal server error\r\n")
|
||||
|
||||
elif data.startswith("LIST ACTIVE") and not data.startswith("LIST ACTIVE.TIMES"):
|
||||
try:
|
||||
pattern = data[11:].strip() or None
|
||||
active_gen = nntp_client.list_active_gen(pattern)
|
||||
self.wfile.write("215 list of newsgroups follows\r\n")
|
||||
for entry in active_gen:
|
||||
self.wfile.write("%s %d %d %s\r\n" % entry)
|
||||
self.wfile.write(".\r\n")
|
||||
self.wfile.write(str(e) + "\r\n")
|
||||
except Exception as ex:
|
||||
print(bcolors.ERROR + str(ex.message) + bcolors.ENDC)
|
||||
self.wfile.write("503 internal server error\r\n")
|
||||
|
||||
elif data.startswith("AUTHINFO user") or data.startswith("AUTHINFO pass"):
|
||||
self.wfile.write("281 Ok\r\n")
|
||||
|
||||
elif data.startswith("XFEATURE"):
|
||||
self.wfile.write("290 feature enabled\r\n")
|
||||
|
||||
elif data.startswith("QUIT"):
|
||||
self.wfile.write("205 Connection closing\r\n")
|
||||
break
|
||||
|
||||
else:
|
||||
self.wfile.write("500 What?\r\n")
|
||||
|
||||
|
||||
# NNTP proxy server for newznab
|
||||
class NNTPProxyServer(SocketServer.ThreadingMixIn, SocketServer.TCPServer):
|
||||
|
||||
allow_reuse_address = True
|
||||
|
||||
def __init__(self, server_address, request_handler, nntp_client_pool_obj, bind_and_activate=True):
|
||||
SocketServer.TCPServer.__init__(self, server_address, request_handler, bind_and_activate=bind_and_activate)
|
||||
self.nntp_client_pool = nntp_client_pool_obj
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
import sys
|
||||
try:
|
||||
if len(sys.argv) == 1:
|
||||
import os
|
||||
pathname = os.path.abspath(os.path.dirname(sys.argv[0]))
|
||||
with open(pathname+"/lib/nntpproxy.conf", "rb") as fd:
|
||||
config = json.load(fd)
|
||||
else:
|
||||
with open(sys.argv[1], "rb") as fd:
|
||||
config = json.load(fd)
|
||||
except IndexError:
|
||||
sys.stderr.write("Usage: %s configfile\n" % sys.argv[0])
|
||||
sys.exit(1)
|
||||
except IOError as e:
|
||||
sys.stderr.write("Failed to open config file (%s)\n" % e)
|
||||
sys.exit(1)
|
||||
except ValueError as e:
|
||||
sys.stderr.write("Failed to parse config file (%s)\n" % e)
|
||||
sys.exit(1)
|
||||
|
||||
nntp_client_pool = socketpool.ConnectionPool(
|
||||
NNTPClientConnector,
|
||||
retry_max=3,
|
||||
retry_delay=1,
|
||||
timeout=-1,
|
||||
max_lifetime=30000.,
|
||||
max_size=int(config["pool"]["size"]),
|
||||
options=config["usenet"]
|
||||
)
|
||||
|
||||
proxy = NNTPProxyServer((config["proxy"]["host"], config["proxy"]["port"]),
|
||||
NNTPProxyRequestHandler, nntp_client_pool)
|
||||
remote = (config["usenet"]["host"], config["usenet"]["port"])
|
||||
print(bcolors.PRIMARY +
|
||||
"NNTP proxy server started on: %s:%d, using a maximum pool size of %d." %
|
||||
(config["proxy"]["host"], config["proxy"]["port"], config["pool"]["size"]))
|
||||
proxy.serve_forever()
|
||||
@@ -0,0 +1,246 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
from __future__ import print_function
|
||||
from __future__ import unicode_literals
|
||||
import sys, os, time
|
||||
import threading
|
||||
try:
|
||||
import queue
|
||||
except ImportError:
|
||||
import Queue as queue
|
||||
import subprocess
|
||||
import string
|
||||
import signal
|
||||
import datetime
|
||||
import lib.info as info
|
||||
from lib.info import bcolors
|
||||
conf = info.readConfig()
|
||||
cur = info.connect()
|
||||
|
||||
if len(sys.argv) == 1:
|
||||
print(bcolors.ERROR + "\nWrong set of arguments.\nThe first argument [additional, nfo, movie, clean] determines the postprocessing to do.\nThe optional second argument for [additional, nfo] [groupid, categoryid] allows to process only that group or category.\nThe optional second argument for [movies, tv] [clean] allows processing only properly renamed releases.\n\npython postprocess_threaded.py [additional, nfo] (optional [groupid, categoryid])\npython postprocess_threaded.py [movie, tv] (optional [clean])\n" + bcolors.ENDC)
|
||||
sys.exit()
|
||||
if len(sys.argv) == 3 and sys.argv[2] == "clean":
|
||||
print(bcolors.HEADER + "\nPostProcess {} Clean Threaded Started at {}".format(sys.argv[1],datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
else:
|
||||
print(bcolors.HEADER + "\nPostProcess {} Threaded Started at {}".format(sys.argv[1],datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
|
||||
if sys.argv[1] == "additional":
|
||||
print(bcolors.HEADER + "Downloaded: b = yEnc article, f= failed ;Processing: z = zip file, r = rar file" + bcolors.ENDC);
|
||||
print(bcolors.HEADER + "Added: s = sample image, j = jpeg image, A = audio sample, a = audio mediainfo, v = video sample" + bcolors.ENDC);
|
||||
print(bcolors.HEADER + "Added: m = video mediainfo, n = nfo, ^ = file details from inside the rar/zip" + bcolors.ENDC);
|
||||
elif sys.argv[1] == "nfo":
|
||||
print(bcolors.HEADER + "* = hidden NFO, + = NFO, - = no NFO, f = download failed." + bcolors.ENDC)
|
||||
|
||||
# You can limit postprocessing for additional and nfo by groupid or categoryid
|
||||
if len(sys.argv) == 3 and sys.argv[2].isdigit() and len(sys.argv[2]) < 4:
|
||||
groupid = 'AND groupid = '+sys.argv[2]
|
||||
print(bcolors.HEADER + "Using groupid "+sys.argv[2] + bcolors.ENDC)
|
||||
elif len(sys.argv) == 3 and sys.argv[2].isdigit() and len(sys.argv[2]) == 4:
|
||||
if sys.argv[2] == '1000':
|
||||
groupid = 'AND categoryid BETWEEN 1000 AND 1999'
|
||||
print(bcolors.HEADER + "Using categoryids 1000-1999" + bcolors.ENDC)
|
||||
elif sys.argv[2] == '2000':
|
||||
groupid = 'AND categoryid BETWEEN 2000 AND 2999'
|
||||
print(bcolors.HEADER + "Using categoryids 2000-2999" + bcolors.ENDC)
|
||||
elif sys.argv[2] == '3000':
|
||||
groupid = 'AND categoryid BETWEEN 3000 AND 3999'
|
||||
print(bcolors.HEADER + "Using categoryids 3000-3999" + bcolors.ENDC)
|
||||
elif sys.argv[2] == '4000':
|
||||
groupid = 'AND categoryid BETWEEN 4000 AND 4999'
|
||||
print(bcolors.HEADER + "Using categoryids 4000-4999" + bcolors.ENDC)
|
||||
elif sys.argv[2] == '5000':
|
||||
groupid = 'AND categoryid BETWEEN 5000 AND 5999'
|
||||
print(bcolors.HEADER + "Using categoryids 5000-5999" + bcolors.ENDC)
|
||||
elif sys.argv[2] == '6000':
|
||||
groupid = 'AND categoryid BETWEEN 6000 AND 6999'
|
||||
print(bcolors.HEADER + "Using categoryids 6000-6999" + bcolors.ENDC)
|
||||
elif sys.argv[2] == '7000':
|
||||
groupid = 'AND categoryid BETWEEN 7000 AND 7999'
|
||||
print(bcolors.HEADER + "Using categoryids 7000-7999" + bcolors.ENDC)
|
||||
elif sys.argv[2] == '8000':
|
||||
groupid = 'AND categoryid BETWEEN 8000 AND 8999'
|
||||
print(bcolors.HEADER + "Using categoryids 8000-8999" + bcolors.ENDC)
|
||||
else:
|
||||
groupid = 'AND categoryid = '+sys.argv[2]
|
||||
print(bcolors.HEADER + "Using categoryid "+sys.argv[2] + bcolors.ENDC)
|
||||
else:
|
||||
groupid = ''
|
||||
|
||||
#you can sort tv releases by searchname
|
||||
if len(sys.argv) == 3 and (sys.argv[2] == "asc" or sys.argv[2] == "desc"):
|
||||
orderBY = 'ORDER BY searchname '+sys.argv[2]
|
||||
print(bcolors.HEADER + "Using ORDER BY searchname "+sys.argv[2] + bcolors.ENDC)
|
||||
if len(sys.argv) == 4 and (sys.argv[3] == "asc" or sys.argv[3] == "desc"):
|
||||
orderBY = 'ORDER BY searchname '+sys.argv[3]
|
||||
print(bcolors.HEADER + "Using CLEAN - ORDER BY searchname "+sys.argv[3] + bcolors.ENDC)
|
||||
else:
|
||||
orderBY = 'ORDER BY postdate DESC'
|
||||
|
||||
start_time = time.time()
|
||||
pathname = os.path.abspath(os.path.dirname(sys.argv[0]))
|
||||
|
||||
if len(sys.argv) > 1 and sys.argv[1] == "additional":
|
||||
cur[0].execute("SELECT (SELECT value FROM site WHERE setting = 'postthreads') AS a, (SELECT value FROM site WHERE setting = 'maxaddprocessed') AS b, (SELECT value FROM site WHERE setting = 'maxnfoprocessed') AS c, (SELECT value FROM site WHERE setting = 'maximdbprocessed') AS d, (SELECT value FROM site WHERE setting = 'maxrageprocessed') AS e, (SELECT value FROM site WHERE setting = 'maxsizetopostprocess') AS f, (SELECT value FROM site WHERE setting = 'tmpunrarpath') AS g, (SELECT value FROM tmux WHERE setting = 'post') AS h, (SELECT value FROM tmux WHERE setting = 'post_non') AS i, (SELECT count(*) FROM releases WHERE haspreview = -1 and passwordstatus = -1 "+groupid+") as j, (SELECT count(*) FROM releases WHERE haspreview = -1 and passwordstatus = -2 "+groupid+") as k, (SELECT count(*) FROM releases WHERE haspreview = -1 and passwordstatus = -3 "+groupid+") as l, (SELECT count(*) FROM releases WHERE haspreview = -1 and passwordstatus = -4 "+groupid+") as m, (SELECT count(*) FROM releases WHERE haspreview = -1 and passwordstatus = -5 "+groupid+") as n, (SELECT count(*) FROM releases WHERE haspreview = -1 and passwordstatus = -6 "+groupid+") as o")
|
||||
dbgrab = cur[0].fetchall()
|
||||
ps1 = format(int(dbgrab[0][9]))
|
||||
ps2 = format(int(dbgrab[0][10]))
|
||||
ps3 = format(int(dbgrab[0][11]))
|
||||
ps4 = format(int(dbgrab[0][12]))
|
||||
ps5 = format(int(dbgrab[0][13]))
|
||||
ps6 = format(int(dbgrab[0][14]))
|
||||
elif len(sys.argv) > 1 and sys.argv[1] == "nfo":
|
||||
cur[0].execute("SELECT (SELECT value FROM site WHERE setting = 'nfothreads') AS a, (SELECT value FROM site WHERE setting = 'maxaddprocessed') AS b, (SELECT value FROM site WHERE setting = 'maxnfoprocessed') AS c, (SELECT value FROM site WHERE setting = 'maximdbprocessed') AS d, (SELECT value FROM site WHERE setting = 'maxrageprocessed') AS e, (SELECT value FROM site WHERE setting = 'maxsizetopostprocess') AS f, (SELECT value FROM site WHERE setting = 'tmpunrarpath') AS g, (SELECT value FROM tmux WHERE setting = 'post') AS h, (SELECT value FROM tmux WHERE setting = 'post_non') AS i, (SELECT count(*) FROM releases WHERE nfostatus = -1 "+groupid+") as j, (SELECT count(*) FROM releases WHERE nfostatus = -2 "+groupid+") as k, (SELECT count(*) FROM releases WHERE nfostatus = -3 "+groupid+") as l, (SELECT count(*) FROM releases WHERE nfostatus = -4 "+groupid+") as m, (SELECT count(*) FROM releases WHERE nfostatus = -5 "+groupid+") as n, (SELECT count(*) FROM releases WHERE nfostatus = -6 "+groupid+") as o")
|
||||
dbgrab = cur[0].fetchall()
|
||||
ps1 = format(int(dbgrab[0][9]))
|
||||
ps2 = format(int(dbgrab[0][10]))
|
||||
ps3 = format(int(dbgrab[0][11]))
|
||||
ps4 = format(int(dbgrab[0][12]))
|
||||
ps5 = format(int(dbgrab[0][13]))
|
||||
ps6 = format(int(dbgrab[0][14]))
|
||||
elif len(sys.argv) > 1 and (sys.argv[1] == "movie" or sys.argv[1] == "tv"):
|
||||
cur[0].execute("SELECT(SELECT value FROM site WHERE setting = 'postthreadsnon') AS a, (SELECT value FROM site WHERE setting = 'maxaddprocessed') AS b, (SELECT value FROM site WHERE setting = 'maxnfoprocessed') AS c, (SELECT value FROM site WHERE setting = 'maximdbprocessed') AS d, (SELECT value FROM site WHERE setting = 'maxrageprocessed') AS e, (SELECT value FROM site WHERE setting = 'maxsizetopostprocess') AS f, (SELECT value FROM site WHERE setting = 'tmpunrarpath') AS g, (SELECT value FROM tmux WHERE setting = 'post') AS h, (SELECT value FROM tmux WHERE setting = 'post_non') AS i")
|
||||
dbgrab = cur[0].fetchall()
|
||||
else:
|
||||
print(bcolors.ERROR + "\nAn argument is required, \npostprocess_threaded.py [additional, nfo, movie, tv]\n" + bcolors.ENDC)
|
||||
sys.exit()
|
||||
|
||||
run_threads = int(dbgrab[0][0])
|
||||
ppperrun = int(dbgrab[0][1])
|
||||
nfoperrun = int(dbgrab[0][2])
|
||||
movieperrun = int(dbgrab[0][3])
|
||||
tvrageperrun = int(dbgrab[0][4])
|
||||
maxsizeck = int(dbgrab[0][5])
|
||||
tmppath = dbgrab[0][6]
|
||||
posttorun = int(dbgrab[0][7])
|
||||
postnon = dbgrab[0][8]
|
||||
maxsize = (int(maxsizeck * 1073741824))
|
||||
|
||||
|
||||
if maxsize == 0:
|
||||
maxsize = ''
|
||||
else:
|
||||
maxsize = 'AND r.size < '+str(maxsizeck * 1073741824)
|
||||
|
||||
datas = []
|
||||
maxtries = -1
|
||||
|
||||
process_additional = run_threads * ppperrun
|
||||
process_nfo = run_threads * nfoperrun
|
||||
|
||||
if sys.argv[1] == "additional":
|
||||
cur[0].execute("SELECT LEFT(r.guid, 1) FROM releases r LEFT JOIN category c ON c.id = r.categoryid WHERE r.nzbstatus = 1 "+maxsize+" AND (r.haspreview = -1 AND c.disablepreview = 0) AND r.passwordstatus BETWEEN -6 AND -1 GROUP BY LEFT(r.guid, 1) LIMIT 16")
|
||||
datas = cur[0].fetchall()
|
||||
elif sys.argv[1] == "nfo":
|
||||
cur[0].execute("SELECT LEFT(guid, 1) FROM releases WHERE nzbstatus = 1 AND nfostatus BETWEEN -8 AND -1 GROUP BY LEFT(guid, 1) LIMIT 16")
|
||||
datas = cur[0].fetchall()
|
||||
elif sys.argv[1] == "movie" and len(sys.argv) == 3 and sys.argv[2] == "clean":
|
||||
cur[0].execute("SELECT LEFT(guid, 1) FROM releases WHERE nzbstatus = 1 AND isrenamed = 1 AND searchname IS NOT NULL AND imdbid IS NULL AND categoryid BETWEEN 2000 AND 2999 GROUP BY LEFT(guid, 1) "+orderBY+" LIMIT 16")
|
||||
datas = cur[0].fetchall()
|
||||
elif sys.argv[1] == "movie":
|
||||
cur[0].execute("SELECT LEFT(guid, 1) FROM releases WHERE nzbstatus = 1 AND searchname IS NOT NULL AND imdbid IS NULL AND categoryid BETWEEN 2000 AND 2999 GROUP BY LEFT(guid, 1) "+orderBY+" LIMIT 16")
|
||||
datas = cur[0].fetchall()
|
||||
elif sys.argv[1] == "tv" and len(sys.argv) == 3 and sys.argv[2] == "clean":
|
||||
cur[0].execute("SELECT LEFT(guid, 1) FROM releases WHERE nzbstatus = 1 AND isrenamed = 1 AND searchname IS NOT NULL AND rageid = -1 AND categoryid BETWEEN 5000 AND 5999 GROUP BY LEFT(guid, 1) "+orderBY+" LIMIT 16")
|
||||
datas = cur[0].fetchall()
|
||||
elif sys.argv[1] == "tv":
|
||||
cur[0].execute("SELECT LEFT(guid, 1) FROM releases WHERE nzbstatus = 1 AND searchname IS NOT NULL AND rageid = -1 AND categoryid BETWEEN 5000 AND 5999 GROUP BY LEFT(guid, 1) "+orderBY+" LIMIT 16")
|
||||
datas = cur[0].fetchall()
|
||||
|
||||
#close connection to mysql
|
||||
info.disconnect(cur[0], cur[1])
|
||||
|
||||
if not datas:
|
||||
print(bcolors.HEADER + "No Work to Process" + bcolors.ENDC)
|
||||
sys.exit()
|
||||
|
||||
my_queue = queue.Queue()
|
||||
time_of_last_run = time.time()
|
||||
|
||||
class queue_runner(threading.Thread):
|
||||
def __init__(self, my_queue):
|
||||
threading.Thread.__init__(self)
|
||||
self.my_queue = my_queue
|
||||
|
||||
def run(self):
|
||||
global time_of_last_run
|
||||
|
||||
while True:
|
||||
try:
|
||||
my_id = self.my_queue.get(True, 1)
|
||||
except:
|
||||
if time.time() - time_of_last_run > 3:
|
||||
return
|
||||
else:
|
||||
if my_id:
|
||||
time_of_last_run = time.time()
|
||||
subprocess.call(["php", pathname+"/../../multiprocessing/.do_not_run/switch.php", "python pp_"+my_id])
|
||||
time.sleep(.02)
|
||||
self.my_queue.task_done()
|
||||
|
||||
def u(x):
|
||||
if sys.version_info[0] < 3:
|
||||
import codecs
|
||||
return codecs.unicode_escape_decode(x)[0]
|
||||
else:
|
||||
return x
|
||||
|
||||
def main(args):
|
||||
global time_of_last_run
|
||||
time_of_last_run = time.time()
|
||||
|
||||
print(bcolors.HEADER + "We will be using a max of {} threads, a queue of {}.".format(run_threads, "{:,}".format(len(datas))) + bcolors.ENDC)
|
||||
|
||||
time.sleep(2)
|
||||
|
||||
def signal_handler(signal, frame):
|
||||
sys.exit(0)
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
|
||||
if True:
|
||||
#spawn a pool of place worker threads
|
||||
for i in range(run_threads):
|
||||
p = queue_runner(my_queue)
|
||||
p.setDaemon(False)
|
||||
p.start()
|
||||
|
||||
#now load some arbitrary jobs into the queue
|
||||
if sys.argv[1] == "additional":
|
||||
for release in datas:
|
||||
time.sleep(.02)
|
||||
my_queue.put(u("%s %s") % (sys.argv[1], release[0]))
|
||||
elif sys.argv[1] == "nfo":
|
||||
for release in datas:
|
||||
time.sleep(.02)
|
||||
my_queue.put(u("%s %s") % (sys.argv[1], release[0]))
|
||||
elif sys.argv[1] == "movie":
|
||||
for release in datas:
|
||||
time.sleep(.02)
|
||||
my_queue.put(u("%s %s") % (sys.argv[1], release[0]))
|
||||
elif sys.argv[1] == "tv":
|
||||
for release in datas:
|
||||
time.sleep(.02)
|
||||
my_queue.put(u("%s %s") % (sys.argv[1], release[0]))
|
||||
|
||||
my_queue.join()
|
||||
|
||||
if sys.argv[1] == "nfo":
|
||||
cur = info.connect()
|
||||
cur[0].execute("SELECT id from releases WHERE nfostatus <= -6")
|
||||
final = cur[0].fetchall()
|
||||
if len(datas) > 0:
|
||||
for item in final:
|
||||
run = "DELETE FROM releasenfo WHERE nfo IS NULL AND releaseid = %s"
|
||||
cur[0].execute(run, (item[0]))
|
||||
final = cur[0].fetchall()
|
||||
|
||||
#close connection to mysql
|
||||
info.disconnect(cur[0], cur[1])
|
||||
|
||||
print(bcolors.HEADER + "\nPostProcess {} Threaded Completed at {}".format(sys.argv[1],datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
print(bcolors.HEADER + "Running time: {}\n\n".format(str(datetime.timedelta(seconds=time.time() - start_time))) + bcolors.ENDC)
|
||||
|
||||
if __name__ == '__main__':
|
||||
main(sys.argv[1:])
|
||||
@@ -0,0 +1,102 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import sys, os, time
|
||||
import threading
|
||||
try:
|
||||
import queue
|
||||
except ImportError:
|
||||
import Queue as queue
|
||||
import subprocess
|
||||
import string
|
||||
import signal
|
||||
import datetime
|
||||
|
||||
import lib.info as info
|
||||
from lib.info import bcolors
|
||||
conf = info.readConfig()
|
||||
cur = info.connect()
|
||||
start_time = time.time()
|
||||
pathname = os.path.abspath(os.path.dirname(sys.argv[0]))
|
||||
|
||||
print(bcolors.HEADER + "\nUpdate Releases Threaded Started at {}".format(datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
|
||||
cur[0].execute("SELECT (SELECT value FROM site WHERE setting = 'tablepergroup') AS a, (SELECT value FROM site WHERE setting = 'releasethreads') AS b")
|
||||
dbgrab = cur[0].fetchall()
|
||||
allowed = int(dbgrab[0][0])
|
||||
threads = int(dbgrab[0][1])
|
||||
if allowed == 0:
|
||||
print(bcolors.ERROR + "Table per group not enabled" + bcolors.ENDC)
|
||||
info.disconnect(cur[0], cur[1])
|
||||
sys.exit()
|
||||
|
||||
cur[0].execute("SELECT table_name FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = '"+conf['DB_NAME']+"' AND table_rows > 0 AND table_name LIKE 'binaries_%'")
|
||||
datas = cur[0].fetchall()
|
||||
|
||||
#close connection to mysql
|
||||
info.disconnect(cur[0], cur[1])
|
||||
|
||||
if not datas:
|
||||
print(bcolors.HEADER + "No Work to Process" + bcolors.ENDC)
|
||||
sys.exit()
|
||||
|
||||
my_queue = queue.Queue()
|
||||
time_of_last_run = time.time()
|
||||
|
||||
class queue_runner(threading.Thread):
|
||||
def __init__(self, my_queue):
|
||||
threading.Thread.__init__(self)
|
||||
self.my_queue = my_queue
|
||||
|
||||
def run(self):
|
||||
global time_of_last_run
|
||||
|
||||
while True:
|
||||
try:
|
||||
my_id = self.my_queue.get(True, 1)
|
||||
except:
|
||||
if time.time() - time_of_last_run > 3:
|
||||
return
|
||||
else:
|
||||
if my_id:
|
||||
time_of_last_run = time.time()
|
||||
subprocess.call(["php", pathname+"/../../multiprocessing/.do_not_run/switch.php", "releases "+my_id])
|
||||
self.my_queue.task_done()
|
||||
|
||||
def main():
|
||||
global time_of_last_run
|
||||
time_of_last_run = time.time()
|
||||
|
||||
print(bcolors.HEADER + "We will be using a max of {} threads, a queue of {} groups".format(threads, "{:,}".format(len(datas))) + bcolors.ENDC)
|
||||
time.sleep(2)
|
||||
|
||||
def signal_handler(signal, frame):
|
||||
sys.exit(0)
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
|
||||
if True:
|
||||
#spawn a pool of place worker threads
|
||||
for i in range(threads):
|
||||
p = queue_runner(my_queue)
|
||||
p.setDaemon(False)
|
||||
p.start()
|
||||
|
||||
#now load some arbitrary jobs into the queue
|
||||
count = 0
|
||||
for release in datas:
|
||||
if count >= threads:
|
||||
count = 0
|
||||
count += 1
|
||||
my_queue.put("%s %s" % (release[0].replace('binaries_', ''), count))
|
||||
|
||||
my_queue.join()
|
||||
|
||||
#stage7b
|
||||
subprocess.call(["php", pathname+"/../../multiprocessing/.do_not_run/switch.php", "python releases "+str(count)+"_"])
|
||||
|
||||
print(bcolors.HEADER + "\nUpdate Releases Threaded Completed at {}".format(datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
print(bcolors.HEADER + "Running time: {}\n\n".format(str(datetime.timedelta(seconds=time.time() - start_time))) + bcolors.ENDC)
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
@@ -0,0 +1,114 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
from __future__ import print_function
|
||||
import sys, os, time
|
||||
import threading
|
||||
import subprocess
|
||||
import string
|
||||
import signal
|
||||
import datetime
|
||||
import re
|
||||
try:
|
||||
import queue
|
||||
except ImportError:
|
||||
import Queue as queue
|
||||
|
||||
try:
|
||||
import urllib2
|
||||
except ImportError:
|
||||
import urllib.request as urllib2
|
||||
|
||||
import lib.info as info
|
||||
from lib.info import bcolors
|
||||
conf = info.readConfig()
|
||||
cur = info.connect()
|
||||
start_time = time.time()
|
||||
pathname = os.path.abspath(os.path.dirname(sys.argv[0]))
|
||||
threads = 5
|
||||
web = None
|
||||
|
||||
try:
|
||||
import urllib
|
||||
except ImportError:
|
||||
import urllib.request as httplib
|
||||
|
||||
try:
|
||||
r1 = urllib.urlopen("http://reqid.nzedb.com").getcode()
|
||||
if r1 == 200:
|
||||
web = True
|
||||
else:
|
||||
web = False
|
||||
except:
|
||||
web = False
|
||||
|
||||
print(bcolors.HEADER + "\n\nRequestID Threaded Started at {}".format(datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
|
||||
cur[0].execute("SELECT value FROM site WHERE setting = 'request_hours'")
|
||||
dbgrab = cur[0].fetchone()
|
||||
request_hours = str(dbgrab[0])
|
||||
cur[0].execute("SELECT DISTINCT(g.id) FROM releases r INNER JOIN groups g ON r.groupid = g.id WHERE r.nzbstatus = 1 AND r.prehashid = 0 AND r.isrequestid = 1 AND r.reqidstatus in (0, -1) OR (r.reqidstatus = -3 AND r.adddate > NOW() - INTERVAL " + request_hours + " HOUR)")
|
||||
datas = cur[0].fetchall()
|
||||
|
||||
#close connection to mysql
|
||||
info.disconnect(cur[0], cur[1])
|
||||
|
||||
if not datas:
|
||||
print(bcolors.HEADER + "No Work to Process" + bcolors.ENDC)
|
||||
sys.exit()
|
||||
|
||||
my_queue = queue.Queue()
|
||||
time_of_last_run = time.time()
|
||||
|
||||
class queue_runner(threading.Thread):
|
||||
def __init__(self, my_queue):
|
||||
threading.Thread.__init__(self)
|
||||
self.my_queue = my_queue
|
||||
|
||||
def run(self):
|
||||
global time_of_last_run
|
||||
|
||||
while True:
|
||||
try:
|
||||
my_id = self.my_queue.get(True, 1)
|
||||
except:
|
||||
if time.time() - time_of_last_run > 3:
|
||||
return
|
||||
else:
|
||||
if my_id:
|
||||
time_of_last_run = time.time()
|
||||
subprocess.call(["php", pathname+"/../../nix_scripts/multiprocessing/.do_not_run/switch.php", "python requestid "+my_id])
|
||||
time.sleep(.03)
|
||||
self.my_queue.task_done()
|
||||
|
||||
def main():
|
||||
global time_of_last_run
|
||||
time_of_last_run = time.time()
|
||||
|
||||
def signal_handler(signal, frame):
|
||||
sys.exit(0)
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
|
||||
if True:
|
||||
print(bcolors.HEADER + "We will be using a max of {} threads, a queue of {} items".format(threads, "{:,}".format(len(datas))) + bcolors.ENDC)
|
||||
time.sleep(2)
|
||||
|
||||
#spawn a pool of place worker threads
|
||||
for i in range(threads):
|
||||
p = queue_runner(my_queue)
|
||||
p.setDaemon(False)
|
||||
p.start()
|
||||
|
||||
#now load some arbitrary jobs into the queue
|
||||
for release in datas:
|
||||
time.sleep(.03)
|
||||
my_queue.put("%s %s %s %s" % (release[0], release[1], release[2], web))
|
||||
|
||||
my_queue.join()
|
||||
|
||||
print(bcolors.HEADER + "\nRequestID Threaded Completed at {}".format(datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
print(bcolors.HEADER + "Running time: {}\n\n".format(str(datetime.timedelta(seconds=time.time() - start_time))) + bcolors.ENDC)
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
@@ -0,0 +1,104 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import sys, os, time
|
||||
import threading
|
||||
try:
|
||||
import queue
|
||||
except ImportError:
|
||||
import Queue as queue
|
||||
import subprocess
|
||||
import string
|
||||
import signal
|
||||
import datetime
|
||||
|
||||
import lib.info as info
|
||||
from lib.info import bcolors
|
||||
conf = info.readConfig()
|
||||
cur = info.connect()
|
||||
start_time = time.time()
|
||||
pathname = os.path.abspath(os.path.dirname(sys.argv[0]))
|
||||
cur[0].execute("SELECT value FROM site WHERE setting = 'releasethreads'")
|
||||
threads = cur[0].fetchone()
|
||||
threads = int(threads[0])
|
||||
|
||||
print(bcolors.HEADER + "\nUpdate Per Group Threaded Started at {}".format(datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
|
||||
cur[0].execute("SELECT value FROM site WHERE setting = 'tablepergroup'")
|
||||
allowed = cur[0].fetchone()
|
||||
if int(allowed[0]) == 0:
|
||||
print(bcolors.ERROR + "Table per group not enabled" + bcolors.ENDC)
|
||||
info.disconnect(cur[0], cur[1])
|
||||
sys.exit()
|
||||
|
||||
cur[0].execute("SELECT id FROM groups WHERE active = 1 ORDER by cast(last_record as signed) - cast(first_record as signed) DESC")
|
||||
datas = cur[0].fetchall()
|
||||
|
||||
#close connection to mysql
|
||||
info.disconnect(cur[0], cur[1])
|
||||
|
||||
if not datas:
|
||||
print(bcolors.HEADER + "No Work to Process" + bcolors.ENDC)
|
||||
sys.exit()
|
||||
|
||||
my_queue = queue.Queue()
|
||||
time_of_last_run = time.time()
|
||||
|
||||
class queue_runner(threading.Thread):
|
||||
def __init__(self, my_queue):
|
||||
threading.Thread.__init__(self)
|
||||
self.my_queue = my_queue
|
||||
|
||||
def run(self):
|
||||
global time_of_last_run
|
||||
|
||||
while True:
|
||||
try:
|
||||
my_id = self.my_queue.get(True, 1)
|
||||
except:
|
||||
if time.time() - time_of_last_run > 3:
|
||||
return
|
||||
else:
|
||||
if my_id:
|
||||
time_of_last_run = time.time()
|
||||
subprocess.call(["php", pathname+"/../../multiprocessing/.do_not_run/switch.php", "python update_per_group "+my_id])
|
||||
self.my_queue.task_done()
|
||||
|
||||
def main():
|
||||
global time_of_last_run
|
||||
time_of_last_run = time.time()
|
||||
|
||||
print(bcolors.HEADER + "We will be using a max of {} threads, a queue of {} groups".format(threads, "{:,}".format(len(datas))) + bcolors.ENDC)
|
||||
time.sleep(2)
|
||||
|
||||
def signal_handler(signal, frame):
|
||||
sys.exit(0)
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
|
||||
if True:
|
||||
#spawn a pool of place worker threads
|
||||
for i in range(threads):
|
||||
p = queue_runner(my_queue)
|
||||
p.setDaemon(False)
|
||||
p.start()
|
||||
|
||||
#now load some arbitrary jobs into the queue
|
||||
count = 0
|
||||
for release in datas:
|
||||
if count >= threads:
|
||||
count = 0
|
||||
count += 1
|
||||
my_queue.put("%s %s" % (str(release[0]), count))
|
||||
|
||||
my_queue.join()
|
||||
|
||||
#stage7b
|
||||
final = "final"
|
||||
subprocess.call(["php", pathname+"/../../multiprocessing/.do_not_run/switch.php", "python releases "+str(count)+"_"])
|
||||
|
||||
print(bcolors.HEADER + "\nUpdate Releases Threaded Completed at {}".format(datetime.datetime.now().strftime("%H:%M:%S")) + bcolors.ENDC)
|
||||
print(bcolors.HEADER + "Running time: {}\n\n".format(str(datetime.timedelta(seconds=time.time() - start_time))) + bcolors.ENDC)
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Reference in New Issue
Block a user