From 18b0b1bdd0f3b44ce35516c0ea19571ef96fb725 Mon Sep 17 00:00:00 2001
From: smiRaphi <neogt404@gmail.com>
Date: Sat, 25 Jan 2025 22:37:49 +0100
Subject: [PATCH] accept tmd file without extension

---
 utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/utils.py b/utils.py
index a2340c4..2bd12fd 100644
--- a/utils.py
+++ b/utils.py
@@ -587,7 +587,7 @@ def cdn2cia(path, out='', title_ver='', cdn_dev=0, cia_dev=0):
     tmd = ''
     tik = ''
     for i in os.listdir('.'):
-        if i.startswith('tmd.'):
+        if i.startswith('tmd.') or i == 'tmd':
             tmds.append(i)
         elif i == 'cetk':
             tik = i
@@ -597,7 +597,7 @@ def cdn2cia(path, out='', title_ver='', cdn_dev=0, cia_dev=0):
     if len(tmds) == 1: # If only one tmd in CDN dir, use it
         tmd = tmds[0]
     else:
-        tmds.sort(key=lambda h: int(h.split('.')[1]))
+        tmds.sort(key=lambda h: int(os.path.splitext(h)[1].strip('.') or 0))
         if title_ver == '': # If title version not provided, use latest one
             tmd = tmds[-1]
         else: