mirror of
https://github.com/Polprzewodnikowy/SummerCart64.git
synced 2024-11-25 07:06:52 +01:00
24 lines
745 B
Python
24 lines
745 B
Python
|
from io import TextIOWrapper
|
||
|
|
import contextlib
|
||
|
|
import os.path
|
||
|
|
import platform
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
@contextlib.contextmanager
|
||
|
|
def lock_volume(volume: TextIOWrapper):
|
||
|
|
if (os.path.ismount(volume.name)):
|
||
|
|
if (platform.system().startswith("Windows")):
|
||
|
|
import msvcrt
|
||
|
|
import win32file
|
||
|
|
import winioctlcon
|
||
|
|
handle = msvcrt.get_osfhandle(volume.fileno())
|
||
|
|
win32file.DeviceIoControl(handle, winioctlcon.FSCTL_LOCK_VOLUME, None, None)
|
||
|
|
try:
|
||
|
|
yield volume
|
||
|
|
finally:
|
||
|
|
try:
|
||
|
|
volume.flush()
|
||
|
|
finally:
|
||
|
|
win32file.DeviceIoControl(handle, winioctlcon.FSCTL_UNLOCK_VOLUME, None, None)
|