Fix "OSError: Unable to find libfuse" on Windows

mtkclient/Library/Filesystem/mtkdafs.py

@@ -1,38 +1,34 @@
-from fuse import FuseOSError, Operations, LoggingMixIn
-from pathlib import Path
-import logging
-import os
-from stat import S_IFDIR, S_IFLNK, S_IFREG
-from time import time
-import mmap
-import errno
+from stat import S_IFDIR, S_IFREG
 from tempfile import NamedTemporaryFile
+from time import time
+import sys
+import os
+if not os.environ.get('FUSE_LIBRARY_PATH') and os.name == 'nt':
+    os.environ.setdefault('FUSE_LIBRARY_PATH', os.path.join(os.path.dirname(__file__), r"bin\winfsp-%s.dll" % ("x64" if sys.maxsize > 0xffffffff else "x86")))
+from fuse import Operations, LoggingMixIn
+
 
 class MtkDaFS(LoggingMixIn, Operations):
     def __init__(self, da_handler, rw=False):
         self.da_handler = da_handler
         self.rw = rw
-        self.files = {}
-
-        self.files['/'] = dict(
+        self.files = {'/': dict(
             st_mode=(S_IFDIR | 0o555),
             st_ctime=time(),
             st_mtime=time(),
             st_atime=time(),
-            st_nlink=2)
-        self.files['/emmc_user.bin'] = dict(
+            st_nlink=2), '/emmc_user.bin': dict(
             st_mode=(S_IFREG | 0o777) if self.rw else (S_IFREG | 0o555),
             st_ctime=time(),
             st_mtime=time(),
             st_atime=time(),
             st_nlink=2,
-            st_size = self.da_handler.mtk.daloader.daconfig.flashsize)
-        self.files['/partitions'] = dict(
+            st_size=self.da_handler.mtk.daloader.daconfig.flashsize), '/partitions': dict(
             st_mode=(S_IFDIR | 0o555),
             st_ctime=time(),
             st_mtime=time(),
             st_atime=time(),
-            st_nlink=2)
+            st_nlink=2)}
 
         for part in self.da_handler.mtk.daloader.get_partition_data():
             self.files[f'/partitions/{part.name}'] = dict(
@@ -41,35 +37,36 @@ def __init__(self, da_handler, rw=False):
                 st_mtime=time(),
                 st_atime=time(),
                 st_nlink=2,
-                st_size = part.sectors*self.da_handler.mtk.daloader.daconfig.pagesize,
-                offset=part.sector*self.da_handler.mtk.daloader.daconfig.pagesize)
+                st_size=part.sectors * self.da_handler.mtk.daloader.daconfig.pagesize,
+                offset=part.sector * self.da_handler.mtk.daloader.daconfig.pagesize)
 
     def readdir(self, path, fh):
-        return ['.', '..'] + [ x.removeprefix(path).removeprefix('/') for x in self.files if x.startswith(path) and x != path]
+        return ['.', '..'] + [x.removeprefix(path).removeprefix('/') for x in self.files if
+                              x.startswith(path) and x != path]
 
     def read(self, path, size, offset, fh):
-        if size+offset > self.files[path]['st_size']:
+        if size + offset > self.files[path]['st_size']:
             return b''
         file_offset = 0
         if 'offset' in self.files[path]:
             file_offset = self.files[path]['offset']
-        data = self.da_handler.da_ro(start=file_offset+offset, length=size, filename='', parttype=None)
+        data = self.da_handler.da_ro(start=file_offset + offset, length=size, filename='', parttype=None)
         return bytes(data)
 
     def write(self, path, data, offset, fh):
         if not self.rw:
             return 0
-            
-        if offset+len(data) > self.files[path]['st_size']:
+
+        if offset + len(data) > self.files[path]['st_size']:
             return b''
-        
+
         file_offset = 0
         if 'offset' in self.files[path]:
             file_offset = self.files[path]['offset']
-        
+
         with NamedTemporaryFile('rb+', buffering=0) as f_write:
             f_write.write(data)
-            self.da_handler.da_wo(start=file_offset+offset, length=len(data), filename=f_write.name, parttype=None)
+            self.da_handler.da_wo(start=file_offset + offset, length=len(data), filename=f_write.name, parttype=None)
         return len(data)
 
     def getattr(self, path, fh=None):