winzer/elwig-misc
3
1
Fork 0
Code Pull Requests Activity

Only use non-network drives for get_drives() in export.py

Browse Source
This commit is contained in:
Lorenz Stechauner
2023-05-01 23:50:47 +02:00
parent 8bc5dffab3
commit 0b857539b1
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
wgmaster/export.py
View File

@ -4,9 +4,10 @@
from typing import Tuple, List, Generator, Dict, Any
import argparse
import os
import sys
import string
import datetime
import ctypes
import ctypes.wintypes
import hashlib
import pypyodbc
@ -18,7 +19,11 @@ IGNORED_NAMES = ['Program Files', 'Program Files (x86)', 'AppData']
def get_drives() -> List[str]:
return [d for d in string.ascii_uppercase if os.path.exists(f'{d}:')]
# only return local file systems
length = (ctypes.wintypes.DWORD * 1)()
mpr = ctypes.WinDLL('mpr')
return [d for d in string.ascii_uppercase
if os.path.exists(f'{d}:') and mpr.WNetGetConnectionW(f'{d}:', None, length) != 0xEA]
def search_accdb(base_path: str = None) -> Generator[str, None, None]: