Bhabananda Das commited on
Commit
af661cf
·
1 Parent(s): 93c3814
FileStream/bot/plugins/FileHandlers/callback.py CHANGED
@@ -148,6 +148,7 @@ async def cb_data(bot: Client, update: CallbackQuery):
148
  #file_name = get_name(message)
149
  #name = file_caption if (file_caption) != "None/Unknown" else file_name
150
  print(f"User:{user_id}","FileName :", name)
 
151
  from FileStream.Tools.cleanup import Get_Title_Year
152
  title, year = Get_Title_Year(name)
153
  print(f"** ANY SEARCH :{name} \nFinal :{title} YEAR :{year}")
 
148
  #file_name = get_name(message)
149
  #name = file_caption if (file_caption) != "None/Unknown" else file_name
150
  print(f"User:{user_id}","FileName :", name)
151
+
152
  from FileStream.Tools.cleanup import Get_Title_Year
153
  title, year = Get_Title_Year(name)
154
  print(f"** ANY SEARCH :{name} \nFinal :{title} YEAR :{year}")
FileStream/server/API/__init__.py CHANGED
@@ -12,6 +12,7 @@ from aiohttp import web
12
  #)
13
  from .downloads import stream_handler
14
  from .uploads import upload_file
 
15
 
16
 
17
  #-----------------------------Functions------------------------#
@@ -33,6 +34,14 @@ cors = aiohttp_cors.setup(api, defaults={"*": aiohttp_cors.ResourceOptions(
33
 
34
  cors.add(api.router.add_get('/', handle_v2))
35
 
 
 
 
 
 
 
 
 
36
  #api.router.add_get('/files', list_all_files_db)
37
  #api.router.add_get('/files/mix', list_all_files)
38
  #api.router.add_get('/tmdb/mix', list_all_files_tmdb)
 
12
  #)
13
  from .downloads import stream_handler
14
  from .uploads import upload_file
15
+ from .search import searchAll
16
 
17
 
18
  #-----------------------------Functions------------------------#
 
34
 
35
  cors.add(api.router.add_get('/', handle_v2))
36
 
37
+
38
+ #serch PI
39
+ api.router.add_get('/search ', searchAll)
40
+ #Movies
41
+ #Webseries
42
+ #trending
43
+ #detils
44
+
45
  #api.router.add_get('/files', list_all_files_db)
46
  #api.router.add_get('/files/mix', list_all_files)
47
  #api.router.add_get('/tmdb/mix', list_all_files_tmdb)
FileStream/server/API/search.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+
3
+ import os
4
+ import json
5
+ import logging
6
+ import asyncio
7
+ import traceback
8
+
9
+ from aiohttp import web
10
+ from pyrogram import raw
11
+ from aiohttp.http_exceptions import BadStatusLine
12
+
13
+ from FileStream.utils.FileProcessors.human_readable import humanbytes
14
+ from FileStream.Database import Database
15
+ from FileStream.config import Telegram, Server
16
+ db = Database(Telegram.DATABASE_URL, Telegram.SESSION_NAME)
17
+
18
+ async def searchAll(request: web.Request):
19
+ data = await request.get()
20
+ query = data.get('query')
21
+ offset = data.get('offset')
22
+ results = []
23
+ offset = int(offset or 0)
24
+ files, next_offset = await db.get_search_results(query,file_type=None,max_results=10,offset=offset)
25
+ for file in files:
26
+ results.append(
27
+ {
28
+ "title": file['file']['file_name'],
29
+ "document_file_id" :file['file']['file_id'],
30
+ "caption" : file['file']['file_name'] or "",
31
+ "description": f"Size: {humanbytes(file['file']['file_size'])}\nType:{file['file']['mime_type']} ",
32
+ }
33
+ )
34
+ return {results,next_offset}