mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2024-11-14 09:30:21 -08:00
76 lines
2.6 KiB
Python
76 lines
2.6 KiB
Python
# Linktastic Module
|
|
# - A python2/3 compatible module that can create hardlinks/symlinks on windows-based systems
|
|
#
|
|
# Linktastic is distributed under the MIT License. The follow are the terms and conditions of using Linktastic.
|
|
#
|
|
# The MIT License (MIT)
|
|
# Copyright (c) 2012 Solipsis Development
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining a copy of this software and
|
|
# associated documentation files (the "Software"), to deal in the Software without restriction,
|
|
# including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
# and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so,
|
|
# subject to the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included in all copies or substantial
|
|
# portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT
|
|
# LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
|
|
# IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
|
# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
|
|
# SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
|
|
import subprocess
|
|
from subprocess import CalledProcessError
|
|
import os
|
|
|
|
|
|
# Prevent spaces from messing with us!
|
|
def _escape_param(param):
|
|
return '"%s"' % param
|
|
|
|
|
|
# Private function to create link on nt-based systems
|
|
def _link_windows(src, dest):
|
|
try:
|
|
subprocess.check_output(
|
|
'cmd /C mklink /H %s %s' % (_escape_param(dest), _escape_param(src)),
|
|
stderr=subprocess.STDOUT)
|
|
except CalledProcessError as err:
|
|
raise IOError(err.output.decode('utf-8'))
|
|
|
|
# TODO, find out what kind of messages Windows sends us from mklink
|
|
# print(stdout)
|
|
# assume if they ret-coded 0 we're good
|
|
|
|
|
|
def _symlink_windows(src, dest):
|
|
try:
|
|
subprocess.check_output(
|
|
'cmd /C mklink %s %s' % (_escape_param(dest), _escape_param(src)),
|
|
stderr=subprocess.STDOUT)
|
|
except CalledProcessError as err:
|
|
raise IOError(err.output.decode('utf-8'))
|
|
|
|
# TODO, find out what kind of messages Windows sends us from mklink
|
|
# print(stdout)
|
|
# assume if they ret-coded 0 we're good
|
|
|
|
|
|
# Create a hard link to src named as dest
|
|
# This version of link, unlike os.link, supports nt systems as well
|
|
def link(src, dest):
|
|
if os.name == 'nt':
|
|
_link_windows(src, dest)
|
|
else:
|
|
os.link(src, dest)
|
|
|
|
|
|
# Create a symlink to src named as dest, but don't fail if you're on nt
|
|
def symlink(src, dest):
|
|
if os.name == 'nt':
|
|
_symlink_windows(src, dest)
|
|
else:
|
|
os.symlink(src, dest)
|