def handleItemForm(request, form, item):
error = None
language = Language.getOrCreate(form.cleaned_data['language'])
source = Source.getOrCreate(form.cleaned_data['source'])
filesize = None
if form.cleaned_data['size']:
filesize = form.cleaned_data['size']
itemInstance = ItemInstance.getOrCreate(size=filesize,
language=language,
source=source,
creator=request.user,
item=item)
itemInstance.save()
if form.cleaned_data['file']:
itemInstance.path = generateDestinationPath(form)
if form.cleaned_data['tar']:
tf = tarfile.open(fileobj=form.cleaned_data['file'])
tf.extractall(path=path)
else:
destination = open(path+os.path.basename(form.cleaned_data['file'].name), 'wb+')
for chunk in form.cleaned_data['file'].chunks():
destination.write(chunk)
destination.close()
itemInstance.size = form.cleaned_data['file'].size
itemInstance.mtime = int(os.stat(path).st_mtime)
itemInstance.present = True
itemInstance.save()
form = None
error = 'Thank you for contributing.'
elif (form.cleaned_data['dlLinks'] or form.cleaned_data['dlLinksFile']) and form.cleaned_data['keyfile']:
dlUrls = []
if form.cleaned_data['dlLinks']:
dlUrls = [ url.strip() for url in form.cleaned_data['dlLinks'].split(',')]
else:
dlUrls = [ url.strip() for url in form.cleaned_data['dlLinksFile'] ]
if len(dlUrls) > 0:
key = EncryptionKey.fromFileHandle(form.cleaned_data['keyfile'])
downloadFileGroup = DownloadFileGroup(itemInstance=itemInstance, key=key)
downloadFileGroup.save()
for url in dlUrls:
df = DownloadFile(downloadFileGroup=downloadFileGroup, downloadLink=url)
df.save()
form = None
error = 'Thank you for contributing.'
else:
form = None
error = "Thank you for enlisting your item. You will be contacted when someone requests this item."
return form, error
def addMovies(directory):
# TODO check if still working with named groups
# inforegex = re.compile('^(?P<title>.*?)(?:\s+-\s+(?P<subtitle>.*?))?(?:\s+\((?P<year>\d{4})\))?(?:\s+\[(?P<source>[a-zA-Z0-9!"$%&/()=\\\\?_-]*)\])?$')
inforegex = re.compile(
'^(?P<name>.*?)(?:\s+-\s+(?P<subname>.*?))?(?:\s+\((?P<year>\d{4})\))?(?:\s+\[(?P<source>[a-zA-Z0-9!"$%&/()=\\\\?_-]*)\])?$'
)
if os.path.isdir(directory):
for mdir in os.listdir(directory):
dirname = os.path.basename(mdir)
mpath = directory + "/" + dirname
match = inforegex.match(dirname)
if match:
movieDict = inforegex.match(dirname).groupdict()
mmtime = int(os.stat(mpath).st_mtime)
msource = Source.getOrCreate(movieDict.get("source", None))
movie = Movie.getOrCreate(
name=movieDict["name"], subname=movieDict.get("subname", None), year=movieDict.get("year", None)
)
try:
itemInstance = ItemInstance.objects.get(path=mpath)
if itemInstance.mtime < mmtime:
itemInstance.path = mpath
itemInstance.mtime = mmtime
itemInstance.size = getDirSize(mpath)
itemInstance.save()
except ItemInstance.DoesNotExist:
msize = getDirSize(mpath)
print movieDict
itemInstance = ItemInstance(
item=movie, path=mpath, source=msource, mtime=mmtime, size=msize, present=True
)
itemInstance.save()
else:
print "WARNING: invalid movie directory format '%s'" % dirname
else:
print 'ERROR: no such directory "%s"' % (directory)
def addSeries(directory):
serieregex = re.compile("^(?P<title>.*?)$")
seasonregex = re.compile(
'^s(?:eason)?\s*0*(?P<number>\d+)(?:\s+-\s+(?P<subtitle>.*?))?(?:\s+\((?P<year>\d{4})\))?(?:\s+\[(?P<source>[a-zA-Z0-9!"$%&/()=\\\\?_-]*)\])?(?:\s+\|(?P<language>.*)\|)?$',
re.I,
)
if os.path.isdir(directory):
for serieDir in os.listdir(directory):
seriePath = directory + "/" + serieDir
if os.path.isdir(seriePath):
serieDirName = os.path.basename(serieDir)
serieName = serieregex.match(serieDirName).group(1)
for seasonDir in os.listdir(seriePath):
if os.path.isdir(seriePath + "/" + seasonDir):
seasonDirName = os.path.basename(seasonDir)
match = seasonregex.match(seasonDirName)
if match:
seasonDict = match.groupdict()
seasonNumber = seasonDict["number"]
ssubname = seasonDict.get("subtitle", None)
syear = seasonDict.get("year", None)
seasonPath = directory + "/" + serieDirName + "/" + seasonDirName
serie = Serie.getOrCreate(name=serieName)
smtime = int(os.stat(seasonPath).st_mtime)
ssource = Source.getOrCreate(name=seasonDict.get("source", None))
slanguage = Language.getOrCreate(name=seasonDict.get("language", None))
directoryListing = ""
dirlist = os.listdir(seasonPath)
dirlist.sort()
for d in dirlist:
directoryListing += "%s\n" % d
directoryListing = directoryListing[:-1]
season = Season.getOrCreate(serie=serie, number=seasonNumber, subname=ssubname, year=syear)
try:
itemInstance = ItemInstance.objects.get(path=seasonPath)
if itemInstance.mtime < smtime:
itemInstance.path = seasonPath
itemInstance.mtime = smtime
itemInstance.size = getDirSize(seasonPath)
itemInstance.save()
except ItemInstance.DoesNotExist:
ssize = getDirSize(seasonPath)
print (serieName, seasonNumber, ssubname, syear, ssource, ssize)
itemInstance = ItemInstance(
item=season,
path=seasonPath,
source=ssource,
language=slanguage,
mtime=smtime,
size=ssize,
# directoryListing=directoryListing,
present=True,
)
itemInstance.save()
else:
print "WARNING: invalid season directory format '%s'" % seasonDirName
else:
print "WARNING: season directory contains invalid file '%s'" % seasonDir
else:
print 'WARNING: invalid file in series directory "%s"' % serieDir
else:
print 'ERROR: no such directory "%s"' % (directory)
