def post(request): """Creates a tag object :param name: Name for tag :type name: str :returns: json """ res = Result() name = request.POST.get('name', None) if not name: res.isError = True res.message = "No name given" return JsonResponse(res) tag, created = Tag.objects.get_or_create(name=name) res.isSuccess = True if created: res.message = "Created" res.append(tag.json()) return JsonResponse(res)
def post(request): """Creates a tag object :param name: Name for tag :type name: str :returns: json """ res = Result() name = request.POST.get("name", None) if not name: res.isError = True res.message = "No name given" return JsonResponse(res) tag, created = Tag.objects.get_or_create(name=name) res.isSuccess = True if created: res.message = "Created" res.append(tag.json()) return JsonResponse(res)
def post(request): """Creates a tag object :param name: Name for tag :type name: str :returns: json """ res = Result() data = request.POST or json.loads(request.body)['body'] name = data.get('name', None) if not name: res.isError = True res.message = "No name given" return JsonResponse(res.asDict()) tag, created = Tag.objects.get_or_create(name=name.lower()) if created: res.message = "Created" else: res.isError = True res.append(tag.json()) return JsonResponse(res.asDict())
def group(request, obj_id=None): res = Result() if request.method == "GET": try: group = Group.objects.get(pk=obj_id) res.append(group.json()) except ObjectDoesNotExist as err: res.isError = True res.message = str(err) elif request.method == "POST": data = json.loads(request.body)["body"] user = getUser(request) if user: items = getObjectsFromGuids(data["guids"]) gallery = data.get("gallery") g = Group() g.author = user g.title = data.get("title", items[0].title) g.thumbnail = items[0].thumbnail g.description = data.get("description", items[0].description) g.save() g.guid = g.getGuid().guid g.save() if gallery: Gallery.objects.get(pk=gallery).groups.add(g) for item in items: g.appendChild(item) res.append(g.json()) else: res.isError = True res.message = "No user found to create group" elif request.method == "PUT": data = json.loads(request.body)["body"] g = Group.objects.get(pk=obj_id) action = data["action"] index = data.get("index") item = Piece.fromGuid(data["guid"]) if action == "append": g.appendChild(item) elif action == "insert": g.insertChild(index, item) else: g.removeChild(item) res.append(g.json()) else: g = Group.objects.get(pk=obj_id) g.delete() return JsonResponse(res.asDict())
def login_(request): data = request.POST or json.loads(request.body)['body'] email = data['email'].lower() username = email.split('@')[0] password = data.get('password') result = Result() user = None if email: user = authenticate(username=username, password=password) else: result.message = 'Please enter an email address' result.isError = True if user: first_name = request.POST.get('first_name') last_name = request.POST.get('last_name') if first_name: user.first_name = first_name if last_name: user.last_name = last_name if not user.is_active: result.message = 'User account not active' result.isError = True else: result.message = 'Invalid Credentials' result.isError = True if result.isError: if request.is_ajax(): return JsonResponse(result.asDict()) else: return render(request, INDEX_HTML, result.asDict()) login(request, user) # -- Create an artist tag for them Tag.objects.get_or_create( name=user.get_full_name(), defaults={'artist': True} ) # -- Create their personal gallery Gallery.objects.get_or_create(title=user.username, defaults={'owner': user, 'security': Gallery.PERSONAL}) if request.is_ajax(): return JsonResponse(result.asDict()) return HttpResponseRedirect('/frog/gallery/1')
def switchArtist(request): data = request.POST or json.loads(request.body)['body'] artist = data.get('artist', None) guids = data.get('guids', '').split(',') res = Result() if artist: if isinstance(artist, int): author = User.objects.get(pk=artist) tag = Tag.objects.get_or_create(name=author.get_full_name().lower(), defaults={'artist': True})[0] else: first, last = artist.lower().split(' ') author = User.objects.get_or_create(first_name=first, last_name=last, defaults={ 'username': '******' % (first[0], last), })[0] tag = Tag.objects.get_or_create(name=artist.lower(), defaults={'artist': True})[0] objects = getObjectsFromGuids(guids) for n in objects: n.author = author n.tagArtist(tag) res.append(userToJson(author)) res.value['tag'] = tag.id else: res.isError = True res.message = "No artist provided" return JsonResponse(res.asDict())
def post(request): res = Result() data = json.loads(request.POST["body"]) tag = Tag.objects.get(name=data['tag']) badge = Badge.objects.get_or_create(tag=tag)[0] if request.FILES.get("image"): incomingfilename = pathlib.Path(request.FILES["image"].name) filename = '{}{}'.format(tag.name, incomingfilename.suffix) dest = getRoot() / "badges" / filename if not dest.parent.exists(): dest.parent.makedirs_p() handle_uploaded_file(dest, request.FILES["image"]) badge.image = "badges/{}".format(filename) if badge: badge.save() res.append(badge.json()) else: res.isError = True res.message = "No badge found" return JsonResponse(res.asDict())
def post(request): """Sets a key to a value on the currently logged in users preferences :param key: Key to set :type key: str :param val: Value to set :type val: primitive :returns: json """ key = request.POST.get('key', None) val = request.POST.get('val', None) res = Result() if key and val: obj, created = UserPref.objects.get_or_create(user=request.user) if created: obj.data = json.dumps(DefaultPrefs.copy()) obj.save() val = json.loads(val) obj.setKey(key, val) obj.save() res.append(obj.json()) res.isSuccess = True else: res.isError = True res.message = 'No key and value provided' return JsonResponse(res)
def post(request): """Returns a serialized object""" data = request.POST or json.loads(request.body)['body'] guid = data.get('guid', None) res = Result() if guid: obj = getObjectsFromGuids([guid,])[0] comment = Comment() comment.comment = data.get('comment', 'No comment') comment.user = request.user comment.user_name = request.user.get_full_name() comment.user_email = request.user.email comment.content_object = obj comment.site_id = get_current_site(request).id comment.save() obj.comment_count += 1 obj.save() emailComment(comment, obj, request) res.append(commentToJson(comment)) else: res.isError = True res.message = "No guid provided" return JsonResponse(res.asDict())
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 post(request): """Returns a serialized object :param obj_id: ID of comment object :type obj_id: int :returns: json """ guid = request.POST.get('guid', None) res = Result() if guid: obj = getObjectsFromGuids([ guid, ])[0] c = Comment() c.comment = request.POST.get('comment', 'No comment') c.user = request.user c.user_name = request.user.get_full_name() c.user_email = request.user.email c.content_object = obj c.site_id = 1 c.save() obj.comment_count = obj.comment_count + 1 obj.save() __email(c, obj) res.append({'id': c.id, 'comment': c.comment}) res.isSuccess = True else: res.isError = True res.message = "No guid provided" return JsonResponse(res)
def post(request): """Sets a key to a value on the currently logged in users preferences :param key: Key to set :type key: str :param val: Value to set :type val: primitive :returns: json """ data = request.POST or json.loads(request.body)['body'] key = data.get('key', None) val = data.get('val', None) res = Result() if key is not None and val is not None: obj, created = UserPref.objects.get_or_create(user=request.user) if created: obj.data = json.dumps(DefaultPrefs.copy()) obj.save() try: val = json.loads(val) except (TypeError, ValueError): pass obj.setKey(key, val) obj.save() res.append(obj.json()) else: res.isError = True res.message = 'No key and value provided' return JsonResponse(res.asDict())
def post(request): """Returns a serialized object :param obj_id: ID of comment object :type obj_id: int :returns: json """ guid = request.POST.get('guid', None) res = Result() if guid: obj = getObjectsFromGuids([guid,])[0] c = Comment() c.comment = request.POST.get('comment', 'No comment') c.user = request.user c.user_name = request.user.get_full_name() c.user_email = request.user.email c.content_object = obj c.site_id = 1 c.save() obj.comment_count = obj.comment_count + 1 obj.save() __email(c, obj) res.append({'id': c.id, 'comment': c.comment}) res.isSuccess = True else: res.isError = True res.message = "No guid provided" return JsonResponse(res)
def post(request): """ Create a Gallery """ defaultname = 'New Gallery %i' % Gallery.objects.all().count() title = request.POST.get('title', defaultname) description = request.POST.get('description', '') security = int(request.POST.get('security', Gallery.PUBLIC)) parentid = request.POST.get('parent') if parentid: parent = Gallery.objects.get(pk=int(parentid)) g, created = parent.gallery_set.get_or_create(title=title) g.security = parent.security else: g, created = Gallery.objects.get_or_create(title=title) g.security = security g.description = description g.owner = request.user g.save() res = Result() res.isSuccess = True res.append(g.json()) res.message = 'Gallery created' if created else '' return JsonResponse(res)
def filterObjects(request, obj_id): """ Filters Gallery for the requested ImageVideo objects. Returns a Result object with serialized objects """ print obj_id obj = Gallery.objects.get(pk=obj_id) if request.user.is_anonymous() and obj.security != Gallery.PUBLIC: res = Result() res.isError = True res.message = 'This gallery is not public' return JsonResponse(res) tags = json.loads(request.GET.get('filters', '[[]]')) rng = request.GET.get('rng', None) more = json.loads(request.GET.get('more', 'false')) models = request.GET.get('models', 'image,video') if models == '': models = 'image,video' tags = filter(None, tags) models = [ContentType.objects.get(app_label='frog', model=x) for x in models.split(',')] return _filter(request, obj, tags=tags, rng=rng, models=models, more=more)
def login_(request): data = request.POST or json.loads(request.body)["body"] email = data["email"].lower() password = data.get("password") result = Result() user = None if email: user = authenticate(username=email, password=password) else: result.message = "Please enter an email address" result.isError = True if user: first_name = request.POST.get("first_name") last_name = request.POST.get("last_name") if first_name: user.first_name = first_name if last_name: user.last_name = last_name if not user.is_active: result.message = "User account not active" result.isError = True else: result.message = "Invalid Credentials" result.isError = True if result.isError: return JsonResponse(result.asDict()) login(request, user) # -- Create an artist tag for them Tag.objects.get_or_create( name=user.get_full_name(), defaults={"artist": True} ) usersgroup = Group.objects.get_or_create(name="users")[0] usersgroup.user_set.add(user) return JsonResponse(result.asDict())
def like(request, guid): obj = Piece.fromGuid(guid) res = Result() if obj.like(request): emailLike(request, obj) else: res.message = 'Cannot "like" things more than once' res.append(obj.json()) return JsonResponse(res.asDict())
def like(request, guid): obj = Piece.fromGuid(guid) res = Result() if obj.like(request): emailLike(request, obj) else: res.isError = True res.message = 'Cannot "like" things more than once' res.append(obj.json()) return JsonResponse(res.asDict())
def login_(request): data = request.POST or json.loads(request.body)['body'] email = data['email'].lower() password = data.get('password') result = Result() user = None if email: user = authenticate(username=email, password=password) else: result.message = 'Please enter an email address' result.isError = True if user: first_name = request.POST.get('first_name') last_name = request.POST.get('last_name') if first_name: user.first_name = first_name if last_name: user.last_name = last_name if not user.is_active: result.message = 'User account not active' result.isError = True else: result.message = 'Invalid Credentials' result.isError = True if result.isError: return JsonResponse(result.asDict()) login(request, user) # -- Create an artist tag for them Tag.objects.get_or_create( name=user.get_full_name(), defaults={'artist': True} ) return JsonResponse(result.asDict())
def login_(request): data = request.POST or json.loads(request.body)['body'] email = data['email'].lower() password = data.get('password') result = Result() user = None if email: user = authenticate(username=email, password=password) else: result.message = 'Please enter an email address' result.isError = True if user: first_name = request.POST.get('first_name') last_name = request.POST.get('last_name') if first_name: user.first_name = first_name if last_name: user.last_name = last_name if not user.is_active: result.message = 'User account not active' result.isError = True else: result.message = 'Invalid Credentials' result.isError = True if result.isError: return JsonResponse(result.asDict()) login(request, user) # -- Create an artist tag for them Tag.objects.get_or_create(name=user.get_full_name(), defaults={'artist': True}) return JsonResponse(result.asDict())
def post(request): """ Create a Gallery """ defaultname = 'New Gallery %i' % Gallery.objects.all().count() data = request.POST or json.loads(request.body)['body'] title = data.get('title', defaultname) description = data.get('description', '') security = int(data.get('security', Gallery.PUBLIC)) g, created = Gallery.objects.get_or_create(title=title) g.security = security g.description = description g.owner = request.user g.save() res = Result() res.append(g.json()) res.message = 'Gallery created' if created else '' return JsonResponse(res.asDict())
def put(request, obj_id): """Updates the content of a comment :param obj_id: ID of comment object :type obj_id: int :returns: json """ res = Result() c = Comment.objects.get(pk=obj_id) content = request.PUT.get('comment', None) if content: c.comment = content c.save() res.append(commentToJson(c)) res.isSuccess = True else: res.isError = True res.message = "No comment provided" return JsonResponse(res)
def put(request, obj_id): """Updates the content of a comment :param obj_id: ID of comment object :type obj_id: int :returns: json """ res = Result() c = Comment.objects.get(pk=obj_id) data = request.PUT or json.loads(request.body)['body'] content = data.get('comment', None) if content: c.comment = content c.save() res.append(commentToJson(c)) else: res.isError = True res.message = "No comment provided" return JsonResponse(res.asDict())
def post(request): """ Create a Gallery """ defaultname = "New Gallery %i" % Gallery.objects.all().count() data = json.loads(request.body)["body"] title = data.get("title", defaultname) description = data.get("description", "") security = int( data.get("security", request.user.frog_prefs.first().clearance)) g, created = Gallery.objects.get_or_create(title=title) g.security = security g.description = description g.owner = request.user g.save() res = Result() res.append(g.json()) res.message = "Gallery created" if created else "" return JsonResponse(res.asDict())
def recordView(request): res = Result() data = json.loads(request.body)["body"] item = getObjectsFromGuids([data["guid"]]) if item: item = item[0] created = ViewRecord.objects.get_or_create( user=request.user, guid=data["guid"] )[1] if created: item.view_count += 1 item.save() res.append(item.view_count) else: res.isError = True res.message = "No object foudn for guid {}".format(data["guid"]) return JsonResponse(res.asDict())
def commentList(request): """Returns a rendered list of comments :returns: html """ if request.method == "POST": return post(request) res = Result() comments = [] guid = request.GET.get("guid", None) if guid: obj = getObjectsFromGuids([guid]) if obj: obj = obj[0] if obj.AssetType == Image.AssetType: model = "image" elif obj.AssetType == Video.AssetType: model = "video" elif obj.AssetType == Group.AssetType: model = "group" elif obj.AssetType == Marmoset.AssetType: model = "marmoset" else: model = None if model: contenttype = ContentType.objects.get(app_label="frog", model=model) comments = Comment.objects.filter(object_pk=obj.id, content_type=contenttype) else: res.isError = True res.message = "Invalid object, could not lookup comments for {}".format( guid) for comment in comments: res.append(commentToJson(comment)) return JsonResponse(res.asDict())
def post(request): """Creates a tag object :param name: Name for tag :type name: str :returns: json """ res = Result() data = json.loads(request.body)["body"] name = data.get("name", None) if not name: res.isError = True res.message = "No name given" return JsonResponse(res.asDict()) tag = Tag.objects.get_or_create(name=name.lower())[0] res.append(tag.json()) return JsonResponse(res.asDict())
def switchArtist(request): artist = request.POST.get('artist', None) guids = request.POST.get('guids', '').split(',') res = Result() if artist: first, last = artist.lower().split(' ') author = User.objects.get_or_create(first_name=first, last_name=last, defaults={ 'username': '******' % (first[0], last), })[0] tag = Tag.objects.get_or_create(name=artist.lower(), defaults={'artist': True})[0] objects = getObjectsFromGuids(guids) for n in objects: n.author = author n.tagArtist(tag) res.isSuccess = True res.append(userToJson(author)) res.value['tag'] = Tag.objects.get(name=artist.lower()).id else: res.isError = True res.message = "No artist provided" return JsonResponse(res)
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') try: username = request.POST.get('user', False) if username: user = User.objects.get(username=username) else: user = request.user uniqueName = request.POST.get( 'uid', models.Piece.getUniqueID(foreignPath, user)) if galleries and models.Gallery.objects.filter( pk__in=[int(g) for g in galleries], uploads=False): raise PermissionDenied() extension = Path(filename).ext.lower() if extension in models.FILE_TYPES['image']: model = models.Image elif extension in models.FILE_TYPES['video']: model = models.Video else: raise MediaTypeError( '{} is not a supported file type'.format(extension)) obj, created = model.objects.get_or_create( unique_id=uniqueName, defaults={'author': user}) guid = obj.getGuid() hashVal = getHashForFile(uploadfile) if hashVal == obj.hash: for gal in galleries: g = models.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 = models.ROOT if models.FROG_PATH: objPath = objPath / models.FROG_PATH objPath = objPath / guid.guid[-2:] / guid.guid / filename hashPath = objPath.parent / hashVal + objPath.ext if not objPath.parent.exists(): objPath.parent.makedirs() handle_uploaded_file(hashPath, uploadfile) obj.hash = hashVal obj.foreign_path = foreignPath obj.title = title or objPath.namebase obj.export(hashVal, hashPath, tags=tags, galleries=galleries) res.append(obj.json()) for key, uploadfile in request.FILES.items(): if key != 'file': dest = objPath.parent / uploadfile.name handle_uploaded_file(dest, uploadfile) 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())
def _filter(request, object_, tags=None, models=(Image, Video), rng=None, more=False): """Filters Piece objects from self based on filters, search, and range :param tags: List of tag IDs to filter :type tags: list :param models: List of model classes to filter on :type models: list :param rng: Range of objects to return. i.e. 0:100 :type rng: str :param more -- bool, Returns more of the same filtered set of images based on session range return list, Objects filtered """ NOW = time.clock() res = Result() idDict = {} objDict = {} lastIDs = {} data = {} LOGGER.debug('init: %f' % (time.clock() - NOW)) if request.user.is_anonymous(): gRange = 300 else: Prefs = json.loads(UserPref.objects.get(user=request.user).data) gRange = Prefs['batchSize'] request.session.setdefault('frog_range', '0:%i' % gRange) if rng: s, e = [int(x) for x in rng.split(':')] else: if more: s = int(request.session.get('frog_range', '0:%i' % gRange).split(':')[1]) e = s + gRange s, e = 0, gRange else: s, e = 0, gRange ## -- Gat all IDs for each model for m in models: indexes = list(m.model_class().objects.all().values_list('id', flat=True)) if not indexes: continue lastIndex = indexes[0] if more: ## -- This is a request for more results idx = request.session.get('last_%s_id' % m.model, lastIndex + 1) lastIDs.setdefault('last_%s_id' % m.model, idx) else: lastIDs['last_%s_id' % m.model] = lastIndex + 1 ## -- Start with objects within range idDict[m.model] = m.model_class().objects.filter(gallery=object_, id__lt=lastIDs['last_%s_id' % m.model]) LOGGER.debug(m.model + '_initial_query: %f' % (time.clock() - NOW)) if tags: for bucket in tags: searchQuery = "" o = None for item in bucket: ## -- filter by tag if isinstance(item, int) or isinstance(item, long): if not o: o = Q() o |= Q(tags__id=item) ## -- add to search string else: searchQuery += item + ' ' if not HAYSTACK: if not o: o = Q() ## -- use a basic search LOGGER.debug('search From LIKE') o |= Q(title__icontains=item) if HAYSTACK and searchQuery != "": ## -- once all tags have been filtered, filter by search searchIDs = search(searchQuery, m.model_class()) if searchIDs: if not o: o = Q() LOGGER.debug('searchFrom haystack:' + str(searchIDs)) o |= Q(id__in=searchIDs) if o: ## -- apply the filters idDict[m.model] = idDict[m.model].filter(o) else: idDict[m.model] = idDict[m.model].none() LOGGER.debug(m.model + '_added_buckets(%i): %f' % (len(tags), time.clock() - NOW)) ## -- Get all ids of filtered objects, this will be a very fast query idDict[m.model] = list(idDict[m.model].values_list('id', flat=True)) LOGGER.debug(m.model + '_queried_ids: %f' % (time.clock() - NOW)) res.message = str(s) + ':' + str(e) ## -- perform the main query to retrieve the objects we want objDict[m.model] = m.model_class().objects.filter(id__in=idDict[m.model]).select_related('author').prefetch_related('tags') if not rng: objDict[m.model] = objDict[m.model][:gRange] objDict[m.model] = list(objDict[m.model]) LOGGER.debug(m.model + '_queried_obj: %f' % (time.clock() - NOW)) ## -- combine and sort all objects by date objects = _sortObjects(**objDict) if len(models) > 1 else objDict.values()[0] objects = objects[s:e] LOGGER.debug('sorted: %f' % (time.clock() - NOW)) ## -- serialize objects for i in objects: for m in models: if isinstance(i, m.model_class()): ## -- set the last ID per model for future lookups lastIDs['last_%s_id' % m.model] = i.id data['last_%s_id' % m.model] = i.id res.append(i.json()) LOGGER.debug('serialized: %f' % (time.clock() - NOW)) request.session['frog_range'] = ':'.join((str(s),str(e))) LOGGER.debug('total: %f' % (time.clock() - NOW)) request.session['last_image_id'] = lastIDs.get('last_image_id', 0) request.session['last_video_id'] = lastIDs.get('last_video_id', 0) data['count'] = len(objects) data['queries'] = connection.queries res.value = data res.isSuccess = True return JsonResponse(res)
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') try: username = request.POST.get('user', False) if username: user = User.objects.get(username=username) else: user = request.user uniqueName = request.POST.get('uid', models.Piece.getUniqueID(foreignPath, user)) if galleries and models.Gallery.objects.filter(pk__in=[int(g) for g in galleries], uploads=False): raise PermissionDenied() extension = Path(filename).ext.lower() if extension in models.FILE_TYPES['image']: model = models.Image elif extension in models.FILE_TYPES['video']: model = models.Video else: raise MediaTypeError('{} is not a supported file type'.format(extension)) obj, created = model.objects.get_or_create(unique_id=uniqueName, defaults={'author': user}) guid = obj.getGuid() hashVal = getHashForFile(uploadfile) if hashVal == obj.hash: for gal in galleries: g = models.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 = models.ROOT if models.FROG_PATH: objPath = objPath / models.FROG_PATH objPath = objPath / guid.guid[-2:] / guid.guid / filename hashPath = objPath.parent / hashVal + objPath.ext if not objPath.parent.exists(): objPath.parent.makedirs() handle_uploaded_file(hashPath, uploadfile) obj.hash = hashVal obj.foreign_path = foreignPath obj.title = title or objPath.namebase obj.export(hashVal, hashPath, tags=tags, galleries=galleries) res.append(obj.json()) for key, uploadfile in request.FILES.items(): if key != 'file': dest = objPath.parent / uploadfile.name handle_uploaded_file(dest, uploadfile) 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())
def post(request, obj): try: data = request.POST or json.loads(request.body)["body"] except RawPostDataException: data = request.POST tags = data.get("tags", "").split(",") resetthumbnail = data.get("reset-thumbnail", False) crop = data.get("crop") res = Result() for tag in tags: try: t = Tag.objects.get(pk=int(tag)) except ValueError: t, created = Tag.objects.get_or_create(name=tag) if created: res.append(t.json()) obj.tags.add(t) if obj.custom_thumbnail and (crop or request.FILES or resetthumbnail): try: os.unlink(getRoot() / obj.custom_thumbnail.name) except OSError: pass if crop: box = [int(_) for _ in crop] # -- Handle thumbnail upload source = Path(obj.source.name) relativedest = obj.getPath(True) / "{:.0f}{}".format( time.time(), source.ext ) dest = getRoot() / relativedest source = getRoot() / source if not dest.parent.exists(): dest.parent.makedirs() source.copy(dest) obj.custom_thumbnail = relativedest image = pilImage.open(dest) # Crop from center image = image.crop(box) image.load() # Resize image.thumbnail( (FROG_THUMB_SIZE, FROG_THUMB_SIZE), pilImage.ANTIALIAS ) image.save(dest) obj.save() if request.FILES: # -- Handle thumbnail upload f = request.FILES.get("file") relativedest = obj.getPath(True) / f.name dest = getRoot() / relativedest handle_uploaded_file(dest, f) obj.custom_thumbnail = relativedest try: if dest.ext == ".psd": image = psd_tools.PSDLoad(dest).as_PIL() else: image = pilImage.open(dest) except IOError as err: res.isError = True res.message = "{} is not a supported thumbnail image type".format( f.name ) return JsonResponse(res.asDict()) box, width, height = cropBox(*image.size) # Resize image.thumbnail((width, height), pilImage.ANTIALIAS) # Crop from center box = cropBox(*image.size)[0] image.crop(box).save(dest) obj.save() if resetthumbnail: obj.custom_thumbnail = None obj.save() res.value = obj.json() return JsonResponse(res.asDict())
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())