From 430dbdbf127168fd147f4bbdb9c20e9b9674a5ef Mon Sep 17 00:00:00 2001
From: vas3k <me@vas3k.ru>
Date: Wed, 26 Feb 2020 10:37:46 +0100
Subject: [PATCH] Use enum instead of strings

---
 parsing/views.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/parsing/views.py b/parsing/views.py
index 02c39cd..15cb938 100644
--- a/parsing/views.py
+++ b/parsing/views.py
@@ -3,17 +3,19 @@ from django.http import Http404
 
 from parsing.exceptions import ParsingException
 from parsing.telegram.telegram import get_channel
+from parsing.telegram.models import MessageType
 
 
 class TelegramChannelFeed(Feed):
-
     FEED_ITEMS = 30
 
     def get_object(self, request, channel):
         limit = int(request.GET.get("size") or self.FEED_ITEMS)
         only = str(request.GET.get("only") or "")
         if only:
-            only = [item.strip() for item in only.split(",")]
+            only = [
+                MessageType.get(item.strip()) for item in only.split(",")
+            ]
             limit = 100  # dirty hack: artificially increase the limit to get more filtered messages
 
         try: