holm/wg-meshconf
1
0
Fork 0
forked from extern/wg-meshconf
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixed import not found issues

Browse Source
This commit is contained in:
K4YT3X 2021-01-12 21:39:58 -05:00
parent 215032f0c3
commit 40872a138a
2 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
wg_meshconf/database_manager.py
View File

@ -19,7 +19,10 @@ with contextlib.suppress(ImportError):
from prettytable import PrettyTable from prettytable import PrettyTable
# local imports # local imports
from .wireguard import WireGuard try:
from wireguard import WireGuard
except ImportError:
from .wireguard import WireGuard
INTERFACE_ATTRIBUTES = [ INTERFACE_ATTRIBUTES = [
"Address", "Address",

10
wg_meshconf/wg_meshconf.py
View File

@ -17,7 +17,10 @@ import pathlib
import sys import sys
# local imports # local imports
from .database_manager import DatabaseManager try:
from database_manager import DatabaseManager
except ImportError:
from .database_manager import DatabaseManager
def parse_arguments(): def parse_arguments():
@ -192,3 +195,8 @@ def main():
"No command specified\nUse wg-meshconf --help to see available commands", "No command specified\nUse wg-meshconf --help to see available commands",
file=sys.stderr, file=sys.stderr,
) )
# launch the main function if it is not imported as a package
if __name__ == "__main__":
main()