99 lines
2.6 KiB
Python
99 lines
2.6 KiB
Python
# THIS SOFTWARE IS A PART OF FREE COMPETITOR PROJECT
|
|
# THE FOLLOWING SOURCE CODE I UNDER THE GNU
|
|
# AGPL LICENSE V3 OR ANY LATER VERSION.
|
|
|
|
# This project is not for simple users, but for
|
|
# web-masters and a like, so we are counting on
|
|
# your ability to set it up and running.
|
|
|
|
import os
|
|
import json
|
|
from difflib import SequenceMatcher # checks how similar are two strings
|
|
|
|
def similar(a, b):
|
|
# I guess it simpifies the syntax for SequenceMatcher
|
|
# In the previous version we use Lavenshtain but it made
|
|
# it an issue for some people to install.
|
|
return SequenceMatcher(None, a, b).ratio()
|
|
|
|
def search_app(name):
|
|
|
|
# This function output a json of an app that's the closest
|
|
# match to the requested name.
|
|
|
|
closest = {}
|
|
match = 0
|
|
|
|
all_apps = []
|
|
for i in os.listdir("apps"):
|
|
if i.endswith(".json"):
|
|
|
|
try:
|
|
with open("apps/"+i) as json_file:
|
|
idata = json.load(json_file)
|
|
except Exception as e:
|
|
print("Error!", i, e)
|
|
idata = {}
|
|
|
|
all_apps.append(idata)
|
|
# Round 1. By the name
|
|
|
|
for i in all_apps:
|
|
for n in i.get("names",[]):
|
|
m = similar(n.lower(), name.lower())
|
|
if m > match and m > 0.49:
|
|
closest = i
|
|
match = m
|
|
|
|
if closest:
|
|
return closest, match
|
|
|
|
# Round 2. By Generic name
|
|
|
|
for i in all_apps:
|
|
for n in i.get("generic_name",[]):
|
|
m = similar(n.lower(), name.lower())
|
|
if m > match:
|
|
closest = i
|
|
match = m
|
|
|
|
if closest:
|
|
return closest, match
|
|
|
|
def suggest(json_data):
|
|
|
|
# This function will suggest
|
|
|
|
|
|
found = []
|
|
all_apps = []
|
|
for i in os.listdir("apps"):
|
|
if i.endswith(".json"):
|
|
|
|
try:
|
|
with open("apps/"+i) as json_file:
|
|
idata = json.load(json_file)
|
|
except Exception as e:
|
|
print("Error!", i, e)
|
|
idata = {}
|
|
|
|
all_apps.append(idata)
|
|
|
|
for i in all_apps:
|
|
score = 0
|
|
for c in ["generic_name", "networks_read", "networks_write", "formats_read", "formats_write"]:
|
|
|
|
for b in json_data.get(c, []):
|
|
if b in i.get(c, []):
|
|
score += 1
|
|
|
|
found.append([score, i])
|
|
|
|
try:
|
|
found.sort(key=lambda x: x[0])
|
|
found = list(reversed(found))
|
|
except Exception as e:
|
|
print("Found problem:", e)
|
|
fount = []
|
|
|
|
return found
|