You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
57 lines
1.3 KiB
57 lines
1.3 KiB
14 years ago
|
from couchpotato.api import addApiView
|
||
|
|
from couchpotato.environment import Env
|
||
|
|
from flask.helpers import jsonify
|
||
|
14 years ago
|
import os
|
||
|
|
import string
|
||
|
|
|
||
|
|
if os.name == 'nt':
|
||
|
|
import win32file
|
||
|
|
|
||
|
|
class FileBrowser():
|
||
|
|
|
||
|
|
def __init__(self, path = '/'):
|
||
|
|
self.path = path
|
||
|
|
|
||
|
14 years ago
|
addApiView('directory.list', self.view)
|
||
|
|
|
||
|
14 years ago
|
def getDirectories(self):
|
||
|
|
|
||
|
|
# Return driveletters or root if path is empty
|
||
|
|
if self.path == '/' or not self.path:
|
||
|
|
if os.name == 'nt':
|
||
|
|
return self.getDriveLetters()
|
||
|
|
self.path = '/'
|
||
|
|
|
||
|
|
dirs = []
|
||
|
|
for f in os.listdir(self.path):
|
||
|
|
path = os.path.join(self.path, f)
|
||
|
|
if(os.path.isdir(path)):
|
||
|
|
dirs.append(path)
|
||
|
|
|
||
|
|
return dirs
|
||
|
|
|
||
|
|
def getFiles(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def getDriveLetters(self):
|
||
|
|
|
||
|
|
driveletters = []
|
||
|
|
for drive in string.ascii_uppercase:
|
||
|
|
if win32file.GetDriveType(drive + ":") == win32file.DRIVE_FIXED:
|
||
|
|
driveletters.append(drive + ":")
|
||
|
|
|
||
|
|
return driveletters
|
||
|
14 years ago
|
|
||
|
|
def view(self):
|
||
|
|
|
||
|
|
try:
|
||
|
|
fb = FileBrowser(Env.getParam('path', '/'))
|
||
|
|
dirs = fb.getDirectories()
|
||
|
|
except:
|
||
|
|
dirs = []
|
||
|
|
|
||
|
|
return jsonify({
|
||
|
|
'empty': len(dirs) == 0,
|
||
|
|
'dirs': dirs,
|
||
|
|
})
|