Update MadApkEndpoint.py

mapadroid/madmin/endpoints/api/apks/MadApkEndpoint.py

@@ -39,6 +39,10 @@ async def get(self):
             except KeyError:
                 return await self._json_response(data=data[apk_type])
 
+    def allowed_file(filename):
+        ALLOWED_EXTENSIONS = set(['apk', 'apkm', 'zip'])
+        return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+
     async def post(self):
         apk_type_raw: str = self.request.match_info.get('apk_type')
         apk_arch_raw: str = self.request.match_info.get('apk_arch')
@@ -58,7 +62,7 @@ async def post(self):
             elif not file.filename:
                 await self._add_notice_message('No file selected for uploading')
                 raise web.HTTPFound(self._url_for("upload"))
-            elif not allowed_file(file.filename):
+            elif not self.allowed_file(file.filename):
                 await self._add_notice_message('Allowed file type is apk only!')
                 raise web.HTTPFound(self._url_for("upload"))
             filename = secure_filename(file.filename)