Escape CIA URLs.

This commit is contained in:
Steveice10 2016-12-16 23:52:11 -08:00
parent e0b86dd612
commit 5f6b3f3cab

View File

@ -35,17 +35,18 @@ qrData = ""
if os.path.isfile(directory): if os.path.isfile(directory):
if directory.endswith(('.cia', '.tik')): if directory.endswith(('.cia', '.tik')):
qrData += baseUrl + os.path.basename(directory) qrData += baseUrl + urllib.parse.quote(os.path.basename(directory))
directory = os.path.dirname(directory) directory = os.path.dirname(directory)
else: else:
for file in [ file for file in next(os.walk(directory))[2] if file.endswith(('.cia', '.tik')) ]: for file in [ file for file in next(os.walk(directory))[2] if file.endswith(('.cia', '.tik')) ]:
qrData += baseUrl + file + "\n" qrData += baseUrl + urllib.parse.quote(file) + "\n"
if len(qrData) == 0: if len(qrData) == 0:
print("No files to serve.") print("No files to serve.")
sys.exit(1) sys.exit(1)
if not directory == "":
os.chdir(directory) os.chdir(directory)
print("") print("")