import os.path

from django.http import HttpResponse
from django.conf import settings
from functools import wraps
from bempy import ImmediateResponse


def returns_blocks(func):
    @wraps(func)
    def wrapper(request):
        page = func(request)
        
        try:
            if isinstance(page, HttpResponse):
                return page
            else:
                response = HttpResponse(page('render',
                                             request=request,
                                             js_filename='bempy.js',
                                             css_filename='bempy.css'))
            return response
        except ImmediateResponse as e:
            return e.response
    wrapper.block = func
    return wrapper

import os.path

from django.http import HttpResponse
from django.conf import settings
from functools import wraps
from bempy import ImmediateResponse


def returns_blocks(func):
    @wraps(func)
    def wrapper(request, *args, **kwargs):
        page = func(request, *args, **kwargs)

        try:
            if isinstance(page, HttpResponse):
                return page
            else:
                response = HttpResponse(page('render',
                                             request=request,
                                             js_filename='bempy.js',
                                             css_filename='bempy.css'))
            return response
        except ImmediateResponse as e:
            return e.response
    wrapper.block = func
    return wrapper

Make wrapper returned by `returns_blocks` decorator, understand additional `args` and `kwargs` arguments.
Make wrapper returned by `returns_blocks` decorator, understand additional `args` and `kwargs` arguments.

  import os.path
  
  from django.http import HttpResponse
  from django.conf import settings
  from functools import wraps
  from bempy import ImmediateResponse
  
  
  def returns_blocks(func):
      @wraps(func)
-     def wrapper(request):
+     def wrapper(request, *args, **kwargs):
-         page = func(request)
+         page = func(request, *args, **kwargs)
-         
+ 
          try:
              if isinstance(page, HttpResponse):
                  return page
              else:
                  response = HttpResponse(page('render',
                                               request=request,
                                               js_filename='bempy.js',
                                               css_filename='bempy.css'))
              return response
          except ImmediateResponse as e:
              return e.response
      wrapper.block = func
      return wrapper
  
Make wrapper returned by `returns_blocks` decorator, understand additional `args` and `kwargs` arguments.
## Code Before:
import os.path

from django.http import HttpResponse
from django.conf import settings
from functools import wraps
from bempy import ImmediateResponse


def returns_blocks(func):
    @wraps(func)
    def wrapper(request):
        page = func(request)
        
        try:
            if isinstance(page, HttpResponse):
                return page
            else:
                response = HttpResponse(page('render',
                                             request=request,
                                             js_filename='bempy.js',
                                             css_filename='bempy.css'))
            return response
        except ImmediateResponse as e:
            return e.response
    wrapper.block = func
    return wrapper

## Instruction:
Make wrapper returned by `returns_blocks` decorator, understand additional `args` and `kwargs` arguments.
## Code After:
import os.path

from django.http import HttpResponse
from django.conf import settings
from functools import wraps
from bempy import ImmediateResponse


def returns_blocks(func):
    @wraps(func)
    def wrapper(request, *args, **kwargs):
        page = func(request, *args, **kwargs)

        try:
            if isinstance(page, HttpResponse):
                return page
            else:
                response = HttpResponse(page('render',
                                             request=request,
                                             js_filename='bempy.js',
                                             css_filename='bempy.css'))
            return response
        except ImmediateResponse as e:
            return e.response
    wrapper.block = func
    return wrapper
