def isUnique(request):
path = request.GET.get('path', None)
res = Result()
if path:
if request.user.is_anonymous():
username = request.GET.get('user', 'noauthor')
user = User.objects.get(username=username)
else:
user = request.user
uniqueID = Piece.getUniqueID(path, user)
img = Image.objects.filter(unique_id=uniqueID)
vid = Video.objects.filter(unique_id=uniqueID)
if img:
res.append(img[0].json())
elif vid:
res.append(vid[0].json())
else:
res.append(True)
res.isSuccess = True
else:
res.isError = True
res.message = "No path provided"
return JsonResponse(res)
def isUnique(request):
path = request.GET.get('path', None)
res = Result()
if path:
if request.user.is_anonymous():
username = request.GET.get('user', 'noauthor')
user = User.objects.get(username=username)
else:
user = request.user
uniqueID = Piece.getUniqueID(path, user)
img = Image.objects.filter(unique_id=uniqueID)
vid = Video.objects.filter(unique_id=uniqueID)
if img:
res.append(img[0].json())
elif vid:
res.append(vid[0].json())
else:
res.append(True)
res.isSuccess = True
else:
res.isError = True
res.message = "No path provided"
return JsonResponse(res)
def isUnique(request):
data = request.POST or json.loads(request.body)["body"]
paths = data.get("paths", [])
res = Result()
if data.get("user"):
user = User.objects.get(username=data["user"])
elif request.user.is_anonymous:
raise HttpResponseForbidden
else:
user = request.user
for path in paths:
uniqueid = Piece.getUniqueID(path, user)
img = Image.objects.filter(unique_id=uniqueid)
vid = Video.objects.filter(unique_id=uniqueid)
if img:
res.append(img[0].json())
elif vid:
res.append(vid[0].json())
else:
res.append(True)
return JsonResponse(res.asDict())
def isUnique(request):
data = request.POST or json.loads(request.body)['body']
paths = data.get('paths', [])
res = Result()
if data.get('user'):
user = User.objects.get(username=data['user'])
elif request.user.is_anonymous():
raise HttpResponseForbidden
else:
user = request.user
for path in paths:
uniqueid = Piece.getUniqueID(path, user)
img = Image.objects.filter(unique_id=uniqueid)
vid = Video.objects.filter(unique_id=uniqueid)
if img:
res.append(img[0].json())
elif vid:
res.append(vid[0].json())
else:
res.append(True)
return JsonResponse(res.asDict())
def handle(self, *args, **options):
if options['topdir'] == '.':
options['topdir'] = options['path']
files = path.Path(options['path']).walk()
extensions = EXT['image'] + EXT['video']
baseuser = User.objects.get_or_create(
username=options['user'],
defaults={'first_name': 'No', 'last_name': 'Author', 'email': '*****@*****.**'}
)[0]
gallery = Gallery.objects.get(pk=1)
for file_ in files:
if options['date']:
d = datetime.datetime.fromtimestamp(file_.mtime)
if d <= options['date']:
self.stdout.write('Skipping {} as it is too old: {}'.format(file_, d))
continue
if file_.ext.lower() in extensions:
try:
user = getAuthor(file_.owner)
except NotImplementedError:
user = baseuser
uniqueName = Piece.getUniqueID(file_, user)
if file_.ext.lower() in EXT['image']:
model = Image
else:
model = Video
obj = model.objects.get_or_create(unique_id=uniqueName, defaults={'author': user})[0]
guid = obj.getGuid()
hashVal = file_.read_hexhash('sha1')
objPath = getRoot() / guid.guid[-2:] / guid.guid / file_.name.lower()
hashPath = objPath.parent / hashVal + objPath.ext
if not objPath.parent.exists():
objPath.parent.makedirs()
file_.copy(hashPath)
obj.hash = hashVal
obj.foreign_path = file_
obj.title = objPath.namebase
obj.created = datetime.datetime.fromtimestamp(file_.mtime)
obj.gallery_set.add(gallery)
tags = []
if options['dirnames']:
tags = file_.parent.replace(options['topdir'], '').replace('\\', '/').split('/')
tags = filter(None, tags)
obj.export(hashVal, hashPath, tags=tags)
self.stdout.write('Added %s\n' % file_)
def handle(self, *args, **options):
if options['topdir'] == '.':
options['topdir'] = args[0]
FILES = Path(args[0]).walk()
ALL_EXT = EXT['image'] + EXT['video']
USER = User.objects.get_or_create(
username='******',
defaults={'first_name': 'No', 'last_name': 'Author', 'email': '*****@*****.**'}
)[0]
GALLERY = Gallery.objects.get(pk=1)
for file_ in FILES:
if file_.ext.lower() in ALL_EXT:
uniqueName = Piece.getUniqueID(file_, USER)
if file_.ext.lower() in EXT['image']:
model = Image
else:
model = Video
obj = model.objects.get_or_create(unique_id=uniqueName, defaults={'author': USER})[0]
guid = obj.getGuid()
hashVal = file_.read_hexhash('sha1')
objPath = getRoot() / guid.guid[-2:] / guid.guid / file_.name.lower()
hashPath = objPath.parent / hashVal + objPath.ext
if not objPath.parent.exists():
objPath.parent.makedirs()
file_.copy(hashPath)
obj.hash = hashVal
obj.foreign_path = file_
obj.title = objPath.namebase
obj.gallery_set.add(GALLERY)
tags = []
if options['dirnames']:
tags = file_.parent.replace(options['topdir'], '').replace('\\', '/').split('/')
tags = filter(None, tags)
obj.export(hashVal, hashPath, tags=tags)
self.stdout.write('Added %s\n' % file_)
def upload(request):
res = Result()
uploadfile = request.FILES.get("file")
if uploadfile:
filename = uploadfile.name
path = request.POST.get("path", None)
if path:
foreignPath = path.replace("'", '"')
else:
foreignPath = filename
galleries = request.POST.get("galleries", "1").split(",")
tags = [
_.strip() for _ in request.POST.get("tags", "").split(",") if _
]
title = request.POST.get("title")
description = request.POST.get("description", "")
force = request.POST.get("force")
try:
username = request.POST.get("user", False)
if username:
user = User.objects.get(username=username)
else:
user = request.user
uniqueName = request.POST.get("uid",
Piece.getUniqueID(foreignPath, user))
if galleries and Gallery.objects.filter(
pk__in=[int(g) for g in galleries], uploads=False):
raise PermissionDenied()
extension = Path(filename).ext.lower()
if extension in FILE_TYPES["image"]:
model = Image
elif extension in FILE_TYPES["video"]:
model = Video
elif extension in FILE_TYPES["marmoset"]:
model = Marmoset
else:
raise MediaTypeError(
"{} is not a supported file type".format(extension))
obj, created = model.objects.get_or_create(unique_id=uniqueName,
defaults={
"author": user,
"hidden": False
})
guid = obj.getGuid()
hashVal = getHashForFile(uploadfile)
if hashVal == obj.hash and not force:
for gal in galleries:
g = Gallery.objects.get(pk=int(gal))
obj.gallery_set.add(g)
res.append(obj.json())
res.message = "Files were the same"
return JsonResponse(res.asDict())
objPath = getRoot() / guid.guid[-2:] / guid.guid / filename
hashPath = objPath.parent / hashVal + objPath.ext
if not objPath.parent.exists():
objPath.parent.makedirs()
# Save uploaded files to asset folder
for key, uploadfile in request.FILES.items():
if key == "file":
handle_uploaded_file(hashPath, uploadfile)
else:
dest = objPath.parent / uploadfile.name
handle_uploaded_file(dest, uploadfile)
if key == "thumbnail":
thumbnail = saveAsPng(dest)
# Resize
image = pilImage.open(thumbnail)
width, height = squareCropDimensions(*image.size)
image.thumbnail((width, height), pilImage.ANTIALIAS)
# Crop from center
box = cropBox(*image.size)
image.crop(box).save(thumbnail)
obj.custom_thumbnail = obj.getPath(
True) / thumbnail.name
obj.save()
obj.hash = hashVal
obj.foreign_path = foreignPath
obj.title = title or objPath.namebase
obj.description = description
obj.export(hashVal, hashPath, tags=tags, galleries=galleries)
res.append(obj.json())
except MediaTypeError as err:
res.isError = True
res.message = str(err)
return JsonResponse(res.asDict())
else:
res.isError = True
res.message = "No file found"
return JsonResponse(res.asDict())
