diff --git a/ereuse_devicehub/inventory/devices/__init__.py b/ereuse_devicehub/inventory/devices/__init__.py deleted file mode 100644 index e69de29b..00000000 diff --git a/ereuse_devicehub/inventory/views.py b/ereuse_devicehub/inventory/views.py index 7b028145..977ad6af 100644 --- a/ereuse_devicehub/inventory/views.py +++ b/ereuse_devicehub/inventory/views.py @@ -4,7 +4,6 @@ from flask.views import View from flask_login import login_required, current_user from ereuse_devicehub.resources.device.models import Device -from ereuse_devicehub.utils import is_safe_url devices = Blueprint('inventory.devices', __name__, url_prefix='/inventory') @@ -14,14 +13,14 @@ class DeviceListView(View): template_name = 'inventory/device_list.html' def dispatch_request(self): - # import pdb; pdb.set_trace() + # TODO @cayop adding filter filter_types = ['Desktop', 'Laptop', 'Server'] devices = Device.query.filter( - Device.owner_id==current_user.id).filter( + Device.owner_id == current_user.id).filter( Device.type.in_(filter_types)) context = {'devices': devices} return flask.render_template(self.template_name, **context) -devices.add_url_rule('/device/list/', view_func=DeviceListView.as_view('devicelist')) +devices.add_url_rule('/device/', view_func=DeviceListView.as_view('devicelist'))