MarioPrzBasto commited on
Commit
2a8236b
verified
1 Parent(s): 074844f

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +23 -1
main.py CHANGED
@@ -90,8 +90,30 @@ async def list_files():
90
  except FileNotFoundError:
91
  raise HTTPException(status_code=404, detail="Diret贸rio de dados n茫o encontrado")
92
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
93
 
94
- @app.get("/files/{origin_id}")
95
  async def get_file_by_origin_id(origin_id: int):
96
  try:
97
  for filename in os.listdir(BASE_DIR):
 
90
  except FileNotFoundError:
91
  raise HTTPException(status_code=404, detail="Diret贸rio de dados n茫o encontrado")
92
 
93
+ @app.get("/files/similar")
94
+ async def list_similar_files():
95
+ try:
96
+ files_data = []
97
+ for filename in os.listdir(BASE_DIR):
98
+ filepath = os.path.join(BASE_DIR, filename)
99
+ if os.path.isfile(filepath):
100
+ try:
101
+ with open(filepath, "r") as f:
102
+ file_content = f.read()
103
+ try:
104
+ file_content_json = json.loads(file_content)
105
+ # Check for similarityOrb and filter
106
+ if "similarityOrb" in file_content_json and file_content_json["similarityOrb"] > 0:
107
+ files_data.append({"filename": filename, "content": file_content_json})
108
+ except json.JSONDecodeError:
109
+ pass # Skip files that are not valid JSON
110
+ except (IOError, OSError) as e:
111
+ raise HTTPException(status_code=500, detail=f"Erro ao ler o ficheiro {filename}: {e}")
112
+ return JSONResponse({"files_data": files_data})
113
+ except FileNotFoundError:
114
+ raise HTTPException(status_code=404, detail="Diret贸rio de dados n茫o encontrado")
115
 
116
+ @app.get("/files/find/{origin_id}")
117
  async def get_file_by_origin_id(origin_id: int):
118
  try:
119
  for filename in os.listdir(BASE_DIR):