daca2: refactoring. gather list of archives from ftp server then disconnect.

This commit is contained in:
Daniel Marjamäki 2013-10-13 14:13:10 +02:00
parent a305698442
commit 554790afcf
1 changed files with 58 additions and 68 deletions

View File

@ -15,6 +15,7 @@ import glob
import os
import socket
import datetime
import time
FTPSERVER = 'ftp.sunet.se'
FTPPATH = '/pub/Linux/distributions/Debian/debian/pool/main/'
@ -30,50 +31,17 @@ def removeAllExceptResults():
os.remove(filename)
def generateDaca2Report(allfolders):
filename = os.path.expanduser('~/daca2/daca2.html')
f = open(filename, 'wt')
f.write('<html>\n')
f.write('<body>\n')
f.write('<h1>DACA2</h1>\n')
def scanarchive(fullpath):
results = open('results.txt', 'at')
results.write(fullpath + '\n')
results.close()
f.write('<h2>All folders</h2>\n')
for folder in allfolders:
results = os.path.expanduser('~/daca2/' + folder + '/results.txt')
if os.path.isfile(results):
f.write(
'<a href="' +
folder +
'/results.txt">' +
folder +
'</a><br>\n')
else:
f.write(folder + '<br>\n')
f.write('</body>\n')
f.write('</html>\n')
f.close()
def scanpackage(package, f):
print('package:' + package)
filename = None
path = FTPPATH + FOLDER + '/' + package
try:
for s in f.nlst(path):
if s[-12:] == '.orig.tar.gz':
filename = s
except socket.error:
pass
except ftplib.error_temp:
pass
if not filename:
return
fullpath = 'ftp://' + FTPSERVER + path + '/' + filename
filename = fullpath[fullpath.rfind('/') + 1:]
subprocess.call(['wget', fullpath])
subprocess.call(['tar', 'xzvf', filename])
subprocess.call(['rm', filename])
if filename[-3:] == '.gz':
subprocess.call(['tar', 'xzvf', filename])
elif filename[-4:] == '.bz2':
subprocess.call(['tar', 'xjvf', filename])
dirname = None
for s in glob.glob(filename[:2] + '*'):
@ -95,13 +63,53 @@ def scanpackage(package, f):
comm = p.communicate()
results = open('results.txt', 'at')
results.write(fullpath + '\n')
results.write(comm[1] + '\n')
results.close()
# remove all files/folders except results.txt
removeAllExceptResults()
if len(sys.argv) == 2:
FOLDER = sys.argv[1]
print('Connect to ' + FTPSERVER)
f = ftplib.FTP(FTPSERVER)
f.login()
print('Get package list in folder ' + FOLDER)
packages = f.nlst(FTPPATH + FOLDER)
archives = []
for package in packages:
filename = None
path = FTPPATH + FOLDER + '/' + package
try:
files = f.nlst(path)
for s in files:
if s.find('.orig.tar.') > 0:
filename = s
if not filename:
for s in files:
if s.find('.tar.') > 0:
filename = s
except socket.error:
print('socket.error')
pass
except ftplib.error_temp:
print('ftplib.error_temp')
pass
except EOFError:
print('EOFError')
pass
if not filename:
print('archive not found for ' + package)
else:
archives.append(package + '/' + filename)
print('Disconnect')
f.quit()
time.sleep(30)
workdir = os.path.expanduser('~/daca2/')
@ -111,40 +119,22 @@ if not os.path.isfile(workdir + 'suppressions.txt'):
suppressions.write('\n')
suppressions.close()
if len(sys.argv) == 2:
FOLDER = sys.argv[1]
print('~/daca2/' + FOLDER)
if not os.path.isdir(workdir + FOLDER):
os.makedirs(workdir + FOLDER)
os.chdir(workdir + FOLDER)
if os.path.isfile('results.txt'):
os.remove('results.txt')
print('Connect to ' + FTPSERVER)
f = ftplib.FTP(FTPSERVER)
f.login()
print('Get package list in folder ' + FOLDER)
packages = f.nlst(FTPPATH + FOLDER)
try:
# remove all files/folders except results.txt
removeAllExceptResults()
results = open('results.txt', 'wt')
results.write('DATE ' + str(datetime.date.today()) + '\n\n')
results.close()
for package in packages:
scanpackage(package, f)
except EOFError:
pass
for archive in archives:
# remove all files/folders except results.txt
removeAllExceptResults()
scanarchive('ftp://' + FTPSERVER + FTPPATH + FOLDER + '/' + archive)
try:
generateDaca2Report(f.nlst(FTPPATH))
except socket.error:
pass
except EOFError:
pass