codex-storage/deluge
1
0
Fork 0
mirror of https://github.com/codex-storage/deluge.git synced 2025-02-06 08:33:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add permission checking to add command(fixing few crashes), sort files by date, place folders before files and ignore files other than .torrent files

Browse Source
This commit is contained in:
Asmageddon 2012-05-26 17:35:29 +02:00
parent 5a33e66c2c
commit fa1a1eb939
1 changed files with 49 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

76
deluge/ui/console/commands/add.py
View File

@ -40,6 +40,7 @@ import deluge.ui.console.colors as colors
from deluge.ui.client import client
import deluge.component as component
import deluge.common
from deluge.ui.common import TorrentInfo
from optparse import make_option
import os
@ -102,6 +103,8 @@ class Command(BaseCommand):
return defer.DeferredList(deferreds)
def complete(self, line):
self.console = component.get("ConsoleUI")
line = line.replace("\ ", " ")
line = os.path.abspath(os.path.expanduser(line))
ret = []
@ -110,40 +113,59 @@ class Command(BaseCommand):
if os.path.isdir(line):
# Directory, so we need to show contents of directory
#ret.extend(os.listdir(line))
for f in os.listdir(line):
# Skip hidden
if f.startswith("."):
continue
f = os.path.join(line, f)
if os.path.isdir(f):
if os.sep == '\\': # Windows path support :|
f += "\\"
else: # \o/ Unix
f += "/"
ret.append(f)
try:
for f in os.listdir(line):
# Skip hidden
if f.startswith("."):
continue
f = os.path.join(line, f)
if os.path.isdir(f):
if os.sep == '\\': # Windows path support :|
f += "\\"
else: # \o/ Unix
f += "/"
elif not f.endswith(".torrent"):
continue
ret.append(f)
except OSError:
self.console.write("{!error!}Permission denied: {!info!}%s" % line)
else:
# This is a file, but we could be looking for another file that
# shares a common prefix.
for f in os.listdir(os.path.dirname(line)):
if f.startswith(os.path.split(line)[1]):
ret.append(os.path.join( os.path.dirname(line), f))
try:
# This is a file, but we could be looking for another file that
# shares a common prefix.
for f in os.listdir(os.path.dirname(line)):
if f.startswith(os.path.split(line)[1]):
ret.append(os.path.join( os.path.dirname(line), f))
except OSError:
self.console.write("{!error!}Permission denied: {!info!}%s" % line)
else:
# This path does not exist, so lets do a listdir on it's parent
# and find any matches.
ret = []
if os.path.isdir(os.path.dirname(line)):
for f in os.listdir(os.path.dirname(line)):
if f.startswith(os.path.split(line)[1]):
p = os.path.join(os.path.dirname(line), f)
try:
ret = []
if os.path.isdir(os.path.dirname(line)):
for f in os.listdir(os.path.dirname(line)):
if f.startswith(os.path.split(line)[1]):
p = os.path.join(os.path.dirname(line), f)
if os.path.isdir(p):
if os.sep == '\\': # Windows path support :|
p += "\\"
else: # \o/ Unix
p += "/"
ret.append(p)
if os.path.isdir(p):
if os.sep == '\\': # Windows path support :|
p += "\\"
else: # \o/ Unix
p += "/"
ret.append(p)
except OSError:
self.console.write("{!error!}Permission denied: {!info!}%s" % line)
#Sort by date instead of whatever we might get from os.listdir
ret = sorted(ret, key=lambda p: os.stat(p).st_mtime, reverse=True)
#Directories first
ret = sorted(ret, key=lambda p: os.path.isdir(p), reverse=True)
#Highlight directory names
for i, filename in enumerate(ret):
if os.path.isdir(filename):
ret[i] = "{!cyan!}%s" % filename
for i in range(0, len(ret)):
ret[i] = ret[i].replace(" ", r"\ ")