from django import template

from quickphotos.models import Photo

register = template.Library()


@register.assignment_tag
def get_latest_photos(user, limit=None):
    photos = Photo.objects.filter(user=user)

    if limit is not None:
        photos = photos[:limit]

    return photos

from django import template

from quickphotos.models import Photo

register = template.Library()


@register.assignment_tag
def get_latest_photos(*args, **kwargs):
    limit = kwargs.pop('limit', None)
    photos = Photo.objects.all()

    if args:
        photos = photos.filter(user__in=args)

    if limit is not None:
        photos = photos[:limit]

    return photos

Add support for multiple users photos
Add support for multiple users photos

  from django import template
  
  from quickphotos.models import Photo
  
  register = template.Library()
  
  
  @register.assignment_tag
- def get_latest_photos(user, limit=None):
+ def get_latest_photos(*args, **kwargs):
+     limit = kwargs.pop('limit', None)
-     photos = Photo.objects.filter(user=user)
+     photos = Photo.objects.all()
+ 
+     if args:
+         photos = photos.filter(user__in=args)
  
      if limit is not None:
          photos = photos[:limit]
  
      return photos
  
Add support for multiple users photos
## Code Before:
from django import template

from quickphotos.models import Photo

register = template.Library()


@register.assignment_tag
def get_latest_photos(user, limit=None):
    photos = Photo.objects.filter(user=user)

    if limit is not None:
        photos = photos[:limit]

    return photos

## Instruction:
Add support for multiple users photos
## Code After:
from django import template

from quickphotos.models import Photo

register = template.Library()


@register.assignment_tag
def get_latest_photos(*args, **kwargs):
    limit = kwargs.pop('limit', None)
    photos = Photo.objects.all()

    if args:
        photos = photos.filter(user__in=args)

    if limit is not None:
        photos = photos[:limit]

    return photos
