"""This function creates a shortcut which can be used to run the CGDAT tool"""
# Import needed modules
import os
import winshell
from win32com.client import Dispatch
import sys
# Import package modules
import cgdat
cgdat_gui_path = cgdat.__file__.replace("\\__init__.py", "")
python_path = sys.executable.replace("\\python.exe", "\\pythonw.exe")
# Get relative script path
DIRNAME = os.path.dirname(os.path.abspath(__file__))
# Set paths
desktop = winshell.desktop()
path = os.path.join(desktop, "CGDAT.lnk")
target = python_path
arguments = '"' + cgdat_gui_path + "\\cgdat_gui.py" + '"'
icon_path = os.path.abspath(os.path.join(DIRNAME, "static/media/CGDAT.ico")).replace(
"\\", "/"
) # Toggle on icon
##############################################################
#### Create shortuct function ####
##############################################################
[docs]def main():
"""Main function that is executed when we use the :samp:`cgdat-shortcut` command. This function creates a
shortcut for the gui on the desktop."""
shell = Dispatch("WScript.Shell")
shortcut = shell.CreateShortCut(path)
shortcut.Targetpath = target
shortcut.Arguments = arguments
shortcut.IconLocation = icon_path
shortcut.save()
main()