def convertOldList2B(utility, indexval):
torrentconfig = utility.torrentconfig
index = str(indexval)
try:
if not torrentconfig.has_section(index):
return False
except:
return False
if indexval == 0:
# backup the old file
oldconfigname = os.path.join(utility.getConfigPath(), "torrent.list")
if existsAndIsReadable(oldconfigname):
try:
copy2(oldconfigname, oldconfigname + ".old")
except:
pass
# Torrent information
filename = torrentconfig.Read("src", section=index)
# Format from earlier 2.7.0 test builds:
if not filename:
# If the src is missing, then we should not try to add the torrent
sys.stdout.write("Filename is empty for index: " + str(index) + "!\n")
return False
elif filename.startswith(utility.getPath()):
src = filename
else:
src = os.path.join(utility.getConfigPath(), "torrent", filename)
filename = os.path.split(src)[1]
newsrc = os.path.join(utility.getConfigPath(), "torrent", filename)
configpath = os.path.join(utility.getConfigPath(), "torrentinfo",
filename + ".info")
config = ConfigReader(configpath, "TorrentInfo")
for name, value in torrentconfig.Items(index):
if name != "src" and value != "":
config.Write(name, value)
config.Flush()
torrentconfig.DeleteGroup(index)
torrentconfig.Write(index, newsrc)
return True
class Utility:
def __init__(self, abcpath, configpath):
self.version = version_id
self.abcpath = abcpath
# Find the directory to save config files, etc.
self.dir_root = configpath
moveOldConfigFiles(self)
self.setupConfig()
# Setup language files
self.lang = Lang(self)
# Convert old INI file
convertINI(self)
# Make torrent directory (if needed)
self.MakeTorrentDir()
self.setupTorrentMakerConfig()
self.setupTorrentList()
self.torrents = {"all": [],
"active": {},
"inactive": {},
"pause": {},
"seeding": {},
"downloading": {}}
self.accessflag = Event()
self.accessflag.set()
self.invalidwinfilenamechar = ''
for i in range(32):
self.invalidwinfilenamechar += chr(i)
self.invalidwinfilenamechar += '"*/:<>?\\|'
self.FILESEM = Semaphore(1)
warned = self.config.Read('torrentassociationwarned', 'int')
if (sys.platform == 'win32' and not warned):
self.regchecker = RegChecker(self)
self.config.Write('torrentassociationwarned', '1')
else:
self.regchecker = None
self.lastdir = {"save": "",
"open": "",
"log": ""}
# Is ABC in the process of shutting down?
self.abcquitting = False
# self.abcdonequitting = False
# Keep track of the last tab that was being viewed
self.lasttab = {"advanced": 0,
"preferences": 0}
self.languages = {}
# Keep track of all the "ManagedList" objects in use
self.lists = {}
self.abcfileframe = None
self.abcbuddyframe = None
def getVersion(self):
return self.version
#===============================================================================
# def getNumPeers(self):
# return self.peer_db.getNumEncounteredPeers()#, self.peer_db.size()
#
# def getNumFiles(self):
# return self.torrent_db.getNumMetadataAndLive()#, self.torrent_db.size()
#===============================================================================
def getConfigPath(self):
return self.dir_root
# TODO: python 2.3.x has a bug with os.access and unicode
# return self.dir_root.decode(sys.getfilesystemencoding())
def setupConfig(self):
defaults = {
# MiscPanel
'language_file': 'english.lang',
'confirmonclose': '1',
'associate': '1',
# DiskPanel
'removetorrent': '0',
'diskfullthreshold': '1',
# RateLimitPanel
# 'maxupload': '5',
'maxuploadrate': '0',
'maxdownloadrate': '0',
'maxseeduploadrate': '0',
# SeedingOptionsPanel
'uploadoption': '0',
'uploadtimeh': '0',
'uploadtimem': '30',
'uploadratio': '100',
# AdvancedNetworkPanel
# AdvancedDiskPanel
# TriblerPanel
'torrentcollectsleep': '15', # for RSS Subscriptions
# VideoPanel
'videoplaybackmode': '0',
# Misc
'enableweb2search': '0',
'torrentassociationwarned': '0',
# GUI
'window_width': '1024',
'window_height': '670',
'detailwindow_width': '800',
'detailwindow_height': '500',
'prefwindow_width': '1000',
'prefwindow_height': '480',
'prefwindow_split': '400',
'sash_position': '-185',
't4t_option': 0, # Seeding items added by Boxun
't4t_ratio': 100, # T4T seeding ratio added by Niels
't4t_hours': 0,
't4t_mins': 30,
'g2g_option': 1,
'g2g_ratio': 75,
'g2g_hours': 0,
'g2g_mins': 30,
'family_filter': 1,
'window_x': "",
'window_y': "",
'use_bundle_magic': 0,
# WebUI
'use_webui': 0,
'webui_port': 8080,
# swift reseed
'swiftreseed': 1
}
if sys.platform == 'win32':
defaults['mintray'] = '2'
# Don't use double quotes here, those are lost when this string is stored in the
# abc.conf file in INI-file format. The code that starts the player will add quotes
# if there is a space in this string.
progfilesdir = os.path.expandvars('${PROGRAMFILES}')
# defaults['videoplayerpath'] = progfilesdir+'\\VideoLAN\\VLC\\vlc.exe'
# Path also valid on MS Vista
defaults['videoplayerpath'] = progfilesdir + '\\Windows Media Player\\wmplayer.exe'
defaults['videoanalyserpath'] = self.getPath() + '\\ffmpeg.exe'
elif sys.platform == 'darwin':
defaults['mintray'] = '0' # tray doesn't make sense on Mac
vlcpath = find_prog_in_PATH("vlc")
if vlcpath is None:
# second try
vlcpath = "/Applications/VLC.app"
if not os.path.exists(vlcpath):
vlcpath = None
if vlcpath is None:
defaults['videoplayerpath'] = "/Applications/QuickTime Player.app"
else:
defaults['videoplayerpath'] = vlcpath
ffmpegpath = find_prog_in_PATH("ffmpeg")
if ffmpegpath is None:
defaults['videoanalyserpath'] = "vlc/ffmpeg"
else:
defaults['videoanalyserpath'] = ffmpegpath
else:
defaults['mintray'] = '0' # Still crashes on Linux sometimes
vlcpath = find_prog_in_PATH("vlc")
if vlcpath is None:
defaults['videoplayerpath'] = "vlc"
else:
defaults['videoplayerpath'] = vlcpath
ffmpegpath = find_prog_in_PATH("ffmpeg")
if ffmpegpath is None:
defaults['videoanalyserpath'] = "ffmpeg"
else:
defaults['videoanalyserpath'] = ffmpegpath
configfilepath = os.path.join(self.getConfigPath(), "abc.conf")
self.config = ConfigReader(configfilepath, "ABC", defaults)
@staticmethod
def _convert__helper_4_1__4_2(abc_config, set_config_func, name, convert=None):
if convert is None:
convert = lambda x: x
if abc_config.Exists(name):
v = abc_config.Read(name)
try:
v = convert(v)
except:
pass
else:
set_config_func(v)
abc_config.DeleteEntry(name)
def setupTorrentMakerConfig(self):
# Arno, 2008-03-27: To keep fileformat compatible
defaults = {
'piece_size': '0', # An index into TorrentMaker.FileInfoPanel.piece_choices
'comment': TorrentDefDefaults['comment'],
'created_by': TorrentDefDefaults['created by'],
'announcedefault': TorrentDefDefaults['announce'],
'announcehistory': '',
'announce-list': TorrentDefDefaults['announce-list'],
'httpseeds': TorrentDefDefaults['httpseeds'],
'makehash_md5': str(TorrentDefDefaults['makehash_md5']),
'makehash_crc32': str(TorrentDefDefaults['makehash_crc32']),
'makehash_sha1': str(TorrentDefDefaults['makehash_sha1']),
'startnow': '1',
'savetorrent': '1',
'createmerkletorrent': '1',
'createtorrentsig': '0',
'useitracker': '1',
'manualtrackerconfig': '0'
}
torrentmakerconfigfilepath = os.path.join(self.getConfigPath(), "maker.conf")
self.makerconfig = ConfigReader(torrentmakerconfigfilepath, "ABC/TorrentMaker", defaults)
def setupTorrentList(self):
torrentfilepath = os.path.join(self.getConfigPath(), "torrent.list")
self.torrentconfig = ConfigReader(torrentfilepath, "list0")
# Initialization that has to be done after the wx.App object
# has been created
def postAppInit(self, iconpath):
try:
self.icon = wx.Icon(iconpath, wx.BITMAP_TYPE_ICO)
except:
pass
# makeActionList(self)
def getLastDir(self, operation="save"):
lastdir = self.lastdir[operation]
if operation == "save":
if not os.access(lastdir, os.F_OK):
lastdir = self.config.Read('defaultfolder')
if not os.access(lastdir, os.F_OK):
lastdir = ""
return lastdir
def setLastDir(self, operation, dir):
self.lastdir[operation] = dir
def getPath(self):
return self.abcpath
# return self.abcpath.decode(sys.getfilesystemencoding())
def getMaxDown(self):
maxdownloadrate = self.config.Read('maxdownloadrate', 'int')
if maxdownloadrate == -1:
return '0'
elif maxdownloadrate == 0:
return 'unlimited'
return str(maxdownloadrate)
def setMaxDown(self, valdown):
if valdown == 'unlimited':
self.config.Write('maxdownloadrate', '0')
elif valdown == '0':
self.config.Write('maxdownloadrate', '-1')
else:
self.config.Write('maxdownloadrate', valdown)
def getMaxUp(self):
maxuploadrate = self.config.Read('maxuploadrate', 'int')
if maxuploadrate == -1:
return '0'
elif maxuploadrate == 0:
return 'unlimited'
return str(maxuploadrate)
def setMaxUp(self, valup):
if valup == 'unlimited':
self.config.Write('maxuploadrate', '0')
self.config.Write('maxseeduploadrate', '0')
elif valup == '0':
self.config.Write('maxuploadrate', '-1')
self.config.Write('maxseeduploadrate', '-1')
else:
self.config.Write('maxuploadrate', valup)
self.config.Write('maxseeduploadrate', valup)
def eta_value(self, n, truncate=3):
if n == -1:
return '<unknown>'
if not n:
return ''
n = int(n)
week, r1 = divmod(n, 60 * 60 * 24 * 7)
day, r2 = divmod(r1, 60 * 60 * 24)
hour, r3 = divmod(r2, 60 * 60)
minute, sec = divmod(r3, 60)
if week > 1000:
return '<unknown>'
weekstr = '%d' % (week) + self.lang.get('l_week')
daystr = '%d' % (day) + self.lang.get('l_day')
hourstr = '%d' % (hour) + self.lang.get('l_hour')
minutestr = '%d' % (minute) + self.lang.get('l_minute')
secstr = '%02d' % (sec) + self.lang.get('l_second')
if week > 0:
text = weekstr
if truncate > 1:
text += ":" + daystr
if truncate > 2:
text += "-" + hourstr
elif day > 0:
text = daystr
if truncate > 1:
text += "-" + hourstr
if truncate > 2:
text += ":" + minutestr
elif hour > 0:
text = hourstr
if truncate > 1:
text += ":" + minutestr
if truncate > 2:
text += ":" + secstr
else:
text = minutestr
if truncate > 1:
text += ":" + secstr
return text
def getMetainfo(self, src, openoptions='rb', style="file"):
return getMetainfo(src, openoptions=openoptions, style=style)
def speed_format(self, s, truncate=1, stopearly=None):
return self.size_format(s, truncate, stopearly) + "/" + self.lang.get('l_second')
def speed_format_new(self, s):
if s != None:
if s < 102400:
text = '%2.1f KB/s' % (s / 1024.0)
elif s < 1022797:
text = '%d KB/s' % (s // 1024)
elif s < 104857600:
text = '%2.1f MB/s' % (s / 1048576.0)
elif s < 1047527425:
text = '%d MB/s' % (s // 1048576)
elif s < 107374182400:
text = '%2.1f GB/s' % (s / 1073741824.0)
elif s < 1072668082177:
text = '%d GB/s' % (s // 1073741824)
else:
text = '%2.1f TB/s' % (s // 1099511627776)
return text
return ''
def size_format(self, s, truncate=None, stopearly=None, applylabel=True, rawsize=False, showbytes=False, labelonly=False, textonly=False):
size = 0.0
label = ""
if truncate is None:
truncate = 2
if ((s < 1024) and showbytes and stopearly is None) or stopearly == "Byte":
truncate = 0
size = s
text = "Byte"
elif ((s < 1048576) and stopearly is None) or stopearly == "KB":
size = (s / 1024.0)
text = "KB"
elif ((s < 1073741824) and stopearly is None) or stopearly == "MB":
size = (s / 1048576.0)
text = "MB"
elif ((s < 1099511627776) and stopearly is None) or stopearly == "GB":
size = (s / 1073741824.0)
text = "GB"
else:
size = (s / 1099511627776.0)
text = "TB"
if textonly:
return text
label = self.lang.get(text)
if labelonly:
return label
if rawsize:
return size
# At this point, only accepting 0, 1, or 2
if truncate == 0:
text = ('%.0f' % size)
elif truncate == 1:
text = ('%.1f' % size)
else:
text = ('%.2f' % size)
if applylabel:
text += ' ' + label
return text
def round_range(self, x):
returnar = set()
for i in range(2500):
value = int(gauss(x, 100))
if value < 0:
continue
diff = abs(value - x)
if diff < 2:
pass
elif diff < 10 and x < 50:
value = int(round(value / 3.0) * 3)
elif diff < 75:
value = int(round(value / 25.0) * 25)
elif diff < 450:
value = int(round(value / 75.0) * 75)
else:
value = int(round(value / 150.0) * 150)
returnar.add(value)
returnar = sorted(returnar)
return returnar
def makeNumCtrl(self, parent, value, integerWidth=6, fractionWidth=0, min=0, max=None, size=wx.DefaultSize):
if size != wx.DefaultSize:
autoSize = False
else:
autoSize = True
return masked.NumCtrl(parent,
value=value,
size=size,
integerWidth=integerWidth,
fractionWidth=fractionWidth,
allowNegative=False,
min=min,
max=max,
groupDigits=False,
useFixedWidthFont=False,
autoSize=autoSize)
def MakeTorrentDir(self):
torrentpath = os.path.join(self.getConfigPath(), "torrent")
pathexists = os.access(torrentpath, os.F_OK)
# If the torrent directory doesn't exist, create it now
if not pathexists:
os.mkdir(torrentpath)
def RemoveEmptyDir(self, basedir, removesubdirs=True):
# remove subdirectories
if removesubdirs:
for root, dirs, files in os.walk(basedir, topdown=False):
for name in dirs:
dirname = os.path.join(root, name)
# Only try to delete if it exists
if os.access(dirname, os.F_OK):
if not os.listdir(dirname):
os.rmdir(dirname)
# remove folder
if os.access(basedir, os.F_OK):
if not os.listdir(basedir):
os.rmdir(basedir)
def makeBitmap(self, bitmap, trans_color=wx.Colour(200, 200, 200)):
button_bmp = wx.Bitmap(os.path.join(self.getPath(), 'icons', bitmap), wx.BITMAP_TYPE_BMP)
button_mask = wx.Mask(button_bmp, trans_color)
button_bmp.SetMask(button_mask)
return button_bmp
def makeBitmapButton(self, parent, bitmap, tooltip, event, trans_color=wx.Colour(200, 200, 200), padx=18, pady=4):
tooltiptext = self.lang.get(tooltip)
button_bmp = self.makeBitmap(bitmap, trans_color)
ID_BUTTON = wx.NewId()
button_btn = wx.BitmapButton(parent, ID_BUTTON, button_bmp, size=wx.Size(button_bmp.GetWidth() + padx, button_bmp.GetHeight() + pady))
button_btn.SetToolTipString(tooltiptext)
parent.Bind(wx.EVT_BUTTON, event, button_btn)
return button_btn
def makeBitmapButtonFit(self, parent, bitmap, tooltip, event, trans_color=wx.Colour(200, 200, 200)):
tooltiptext = self.lang.get(tooltip)
button_bmp = self.makeBitmap(bitmap, trans_color)
ID_BUTTON = wx.NewId()
button_btn = wx.BitmapButton(parent, ID_BUTTON, button_bmp, size=wx.Size(button_bmp.GetWidth(), button_bmp.GetHeight()))
button_btn.SetToolTipString(tooltiptext)
parent.Bind(wx.EVT_BUTTON, event, button_btn)
return button_btn
# Check if str is a valid Windows file name (or unit name if unit is true)
# If the filename isn't valid: returns a fixed name
# If the filename is valid: returns an empty string
def fixWindowsName(self, name, unit=False):
if unit and (len(name) != 2 or name[1] != ':'):
return 'c:'
if not name or name == '.' or name == '..':
return '_'
if unit:
name = name[0]
fixed = False
if len(name) > 250:
name = name[:250]
fixed = True
fixedname = ''
spaces = 0
for c in name:
if c in self.invalidwinfilenamechar:
fixedname += '_'
fixed = True
else:
fixedname += c
if c == ' ':
spaces += 1
if fixed:
return fixedname
elif spaces == len(name):
# contains only spaces
return '_'
else:
return ''
def checkWinPath(self, parent, pathtocheck):
if pathtocheck and pathtocheck[-1] == '\\' and pathtocheck != '\\\\':
pathitems = pathtocheck[:-1].split('\\')
else:
pathitems = pathtocheck.split('\\')
nexttotest = 1
if self.isPathRelative(pathtocheck):
# Relative path
# Empty relative path is allowed
if pathtocheck == '':
return True
fixedname = self.fixWindowsName(pathitems[0])
if fixedname:
dlg = wx.MessageDialog(parent,
pathitems[0] + '\n' +
self.lang.get('invalidwinname') + '\n' +
self.lang.get('suggestedname') + '\n\n' +
fixedname,
self.lang.get('error'), wx.ICON_ERROR)
dlg.ShowModal()
dlg.Destroy()
return False
else:
# Absolute path
# An absolute path must have at least one '\'
if not '\\' in pathtocheck:
dlg = wx.MessageDialog(parent, pathitems[0] + '\n' + self.lang.get('errorinvalidpath'),
self.lang.get('error'), wx.ICON_ERROR)
dlg.ShowModal()
dlg.Destroy()
return False
if pathtocheck[:2] != '\\\\':
# Not a network path
fixedname = self.fixWindowsName(pathitems[0], unit=True)
if fixedname:
dlg = wx.MessageDialog(parent,
pathitems[0] + '\n' +
self.lang.get('invalidwinname') +
fixedname,
self.lang.get('error'), wx.ICON_ERROR)
dlg.ShowModal()
dlg.Destroy()
return False
else:
# Network path
nexttotest = 2
for name in pathitems[nexttotest:]:
fixedname = self.fixWindowsName(name)
if fixedname:
dlg = wx.MessageDialog(parent, name + '\n' + self.lang.get('errorinvalidwinname') + fixedname,
self.lang.get('error'), wx.ICON_ERROR)
dlg.ShowModal()
dlg.Destroy()
return False
return True
def isPathRelative(self, path):
if len(path) < 2 or path[1] != ':' and path[:2] != '\\\\':
return True
return False
# Get a dictionary with information about a font
def getInfoFromFont(self, font):
default = wx.SystemSettings.GetFont(wx.SYS_DEFAULT_GUI_FONT)
try:
if font.Ok():
font_to_use = font
else:
font_to_use = default
fontname = font_to_use.GetFaceName()
fontsize = font_to_use.GetPointSize()
fontstyle = font_to_use.GetStyle()
fontweight = font_to_use.GetWeight()
fontinfo = {'name': fontname,
'size': fontsize,
'style': fontstyle,
'weight': fontweight}
except:
fontinfo = {'name': "",
'size': 8,
'style': wx.FONTSTYLE_NORMAL,
'weight': wx.FONTWEIGHT_NORMAL}
return fontinfo
def getFontFromInfo(self, fontinfo):
size = fontinfo['size']
name = fontinfo['name']
style = fontinfo['style']
weight = fontinfo['weight']
try:
font = wx.Font(size, wx.DEFAULT, style, weight, faceName=name)
except:
font = wx.SystemSettings.GetFont(wx.SYS_DEFAULT_GUI_FONT)
return font
# Make an entry for a popup menu
def makePopup(self, menu, event=None, label="", extralabel="", bindto=None, type="normal", status=""):
text = ""
if label != "":
text = self.lang.get(label)
text += extralabel
newid = wx.NewId()
if event is not None:
if bindto is None:
bindto = menu
bindto.Bind(wx.EVT_MENU, event, id=newid)
if type == "normal":
menu.Append(newid, text)
elif type == "checkitem":
menu.AppendCheckItem(newid, text)
if status == "active":
menu.Check(newid, True)
if event is None:
menu.Enable(newid, False)
return newid
