mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2025-07-06 13:11:11 -07:00
84 lines
3.1 KiB
Python
84 lines
3.1 KiB
Python
import os
|
|
import shutil
|
|
|
|
import linktastic
|
|
|
|
from core import logger
|
|
from core.utils.paths import make_dir
|
|
|
|
if os.name == 'nt':
|
|
import jaraco
|
|
|
|
|
|
def copy_link(src, target_link, use_link):
|
|
logger.info('MEDIAFILE: [{0}]'.format(os.path.basename(target_link)), 'COPYLINK')
|
|
logger.info('SOURCE FOLDER: [{0}]'.format(os.path.dirname(src)), 'COPYLINK')
|
|
logger.info('TARGET FOLDER: [{0}]'.format(os.path.dirname(target_link)), 'COPYLINK')
|
|
|
|
if src != target_link and os.path.exists(target_link):
|
|
logger.info('MEDIAFILE already exists in the TARGET folder, skipping ...', 'COPYLINK')
|
|
return True
|
|
elif src == target_link and os.path.isfile(target_link) and os.path.isfile(src):
|
|
logger.info('SOURCE AND TARGET files are the same, skipping ...', 'COPYLINK')
|
|
return True
|
|
elif src == os.path.dirname(target_link):
|
|
logger.info('SOURCE AND TARGET folders are the same, skipping ...', 'COPYLINK')
|
|
return True
|
|
|
|
make_dir(os.path.dirname(target_link))
|
|
try:
|
|
if use_link == 'dir':
|
|
logger.info('Directory linking SOURCE FOLDER -> TARGET FOLDER', 'COPYLINK')
|
|
linktastic.dirlink(src, target_link)
|
|
return True
|
|
if use_link == 'junction':
|
|
logger.info('Directory junction linking SOURCE FOLDER -> TARGET FOLDER', 'COPYLINK')
|
|
linktastic.dirlink(src, target_link)
|
|
return True
|
|
elif use_link == 'hard':
|
|
logger.info('Hard linking SOURCE MEDIAFILE -> TARGET FOLDER', 'COPYLINK')
|
|
linktastic.link(src, target_link)
|
|
return True
|
|
elif use_link == 'sym':
|
|
logger.info('Sym linking SOURCE MEDIAFILE -> TARGET FOLDER', 'COPYLINK')
|
|
linktastic.symlink(src, target_link)
|
|
return True
|
|
elif use_link == 'move-sym':
|
|
logger.info('Sym linking SOURCE MEDIAFILE -> TARGET FOLDER', 'COPYLINK')
|
|
shutil.move(src, target_link)
|
|
linktastic.symlink(target_link, src)
|
|
return True
|
|
elif use_link == 'move':
|
|
logger.info('Moving SOURCE MEDIAFILE -> TARGET FOLDER', 'COPYLINK')
|
|
shutil.move(src, target_link)
|
|
return True
|
|
except Exception as e:
|
|
logger.warning('Error: {0}, copying instead ... '.format(e), 'COPYLINK')
|
|
|
|
logger.info('Copying SOURCE MEDIAFILE -> TARGET FOLDER', 'COPYLINK')
|
|
shutil.copy(src, target_link)
|
|
|
|
return True
|
|
|
|
|
|
def replace_links(link):
|
|
n = 0
|
|
target = link
|
|
if os.name == 'nt':
|
|
if not jaraco.windows.filesystem.islink(link):
|
|
logger.debug('{0} is not a link'.format(link))
|
|
return
|
|
while jaraco.windows.filesystem.islink(target):
|
|
target = jaraco.windows.filesystem.readlink(target)
|
|
n = n + 1
|
|
else:
|
|
if not os.path.islink(link):
|
|
logger.debug('{0} is not a link'.format(link))
|
|
return
|
|
while os.path.islink(target):
|
|
target = os.readlink(target)
|
|
n = n + 1
|
|
if n > 1:
|
|
logger.info('Changing sym-link: {0} to point directly to file: {1}'.format(link, target), 'COPYLINK')
|
|
os.unlink(link)
|
|
linktastic.symlink(target, link)
|