Fixed error in Windows paths

Brachi · Mar 23, 2016 · 4fae2d4 · 4fae2d4
1 parent ef6ff77
commit 4fae2d4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/albam/mtframework/__init__.py b/albam/mtframework/__init__.py
@@ -27,6 +27,7 @@
  'ev616_00.arc',
  'ev617_00.arc',
  'ev618_00.arc',
+ 'uOm09882.arc',
  's109.arc',
  's119.arc',
  's300.arc',

diff --git a/albam/mtframework/arc.py b/albam/mtframework/arc.py
@@ -1,7 +1,6 @@
 from ctypes import Structure, sizeof, c_int, c_uint, c_char, c_short, c_ubyte
 import ntpath
 import os
-import re
 import zlib
 
 from albam.mtframework.mappers import FILE_ID_TO_EXTENSION, EXTENSION_TO_FILE_ID
@@ -99,7 +98,8 @@ def _get_path(file_path, file_type_id, output_path):
 
  @staticmethod
  def _set_path(source_path, file_path):
- file_path = re.sub(source_path + os.path.sep, '', file_path, count=1)
+ source_path = source_path + os.path.sep if not source_path.endswith(os.path.sep) else source_path
+ file_path = file_path.replace(source_path, '')
  file_path = os.path.splitext(file_path)[0]
  parts = file_path.split(os.path.sep)
  return ntpath.join('', *parts).encode('ascii')