Skip to content

Commit 69f677d

Browse files
committed
error with empty db removed
1 parent 41c401f commit 69f677d

File tree

1 file changed

+17
-8
lines changed

1 file changed

+17
-8
lines changed

Link-Preview/linkPreview.py

Lines changed: 17 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -56,18 +56,23 @@ def getImage(soup, url):
5656
res = ogImg or twitterImg or metaImg or img
5757
res = res.get("content", None) or res.get_text() or res.get("src", None)
5858

59+
count = 0
60+
for i in range(0, len(res)):
61+
if (res[i] == "." or res[i] == "/"):
62+
count += 1
63+
else:
64+
break
65+
res = res[count::]
5966
if ((not res == None) and ((not "https://" in res) or (not "https://" in res))):
60-
res.replace(".", "")
61-
if (not res[0] == "/"):
62-
res = "/" + res
63-
res = url + res
67+
res = url + "/" + res
6468
if (res == None):
6569
res = "Not available"
6670

6771
return res
6872

69-
7073
# print dictionary
74+
75+
7176
def printData(data):
7277
print("\nTitle : ", data["title"])
7378
print("Description : ", data["description"])
@@ -95,13 +100,17 @@ def printData(data):
95100
db = {}
96101
# create file if it doesn't exist
97102
if not os.path.exists('Link-Preview/db.json'):
98-
f = open('Link-Preview/db.json', 'w')
103+
f = open('Link-Preview/db.json', "w")
99104
f.write("{}")
100105
f.close()
101106

102107
# read db
103-
with open('Link-Preview/db.json', 'r') as file:
104-
db = json.loads(file.read())
108+
with open('Link-Preview/db.json', 'r+') as file:
109+
data = file.read()
110+
if (len(data) == 0):
111+
data = "{}"
112+
file.write(data)
113+
db = json.loads(data)
105114

106115
# check if it exists
107116
if (url in db):

0 commit comments

Comments
 (0)