def browseSaveLocal(self,saveDict):
filter=gtk.FileFilter()
filter.set_name('template files')
filter.add_pattern('*.tmpl')
dialog = gtk.FileChooserDialog("Open..",
None,
gtk.FILE_CHOOSER_ACTION_SAVE,
(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL,
gtk.STOCK_SAVE, gtk.RESPONSE_OK))
dialog.set_default_response(gtk.RESPONSE_OK)
if not dialog.set_current_folder(os.path.join(get_user_data_dir(),'stats')):
dialog.set_current_folder(get_user_data_dir())
dialog.add_filter(filter)
response = dialog.run()
if response == gtk.RESPONSE_OK:
#print dialog.get_filename(), 'selected'
filename = dialog.get_filename()
elif response == gtk.RESPONSE_CANCEL:
filename=None
print 'Closed, no files selected'
dialog.destroy()
if filename and '.' not in os.path.basename(filename):
filename+='.tmpl'
self.saveTemplateFile(filename,saveDict)
def browseLocal(self):
filter=gtk.FileFilter()
filter.set_name('stat files')
filter.add_pattern('*.db')
dialog = gtk.FileChooserDialog("Open..",
None,
gtk.FILE_CHOOSER_ACTION_OPEN,
(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL,
gtk.STOCK_OPEN, gtk.RESPONSE_OK))
dialog.set_default_response(gtk.RESPONSE_OK)
if not dialog.set_current_folder(os.path.join(get_user_data_dir(),'stats')):
dialog.set_current_folder(get_user_data_dir())
dialog.add_filter(filter)
response = dialog.run()
if response == gtk.RESPONSE_OK:
#print dialog.get_filename(), 'selected'
filename = dialog.get_filename()
elif response == gtk.RESPONSE_CANCEL:
filename=None
print 'Closed, no files selected'
dialog.destroy()
self.browseFinished(filename)
def __init__(self,level='debug',name='p2ner',server=False):
self.log = logging.getLogger(name)
name=name+'.'
self.lname=name
userdatadir = get_user_data_dir()
if not os.path.isdir(userdatadir):
os.mkdir(userdatadir)
if not os.path.isdir(os.path.join(userdatadir, "log")):
os.mkdir(os.path.join(userdatadir, "log"))
self.dir= os.path.join(get_user_data_dir(), "log",name)
self.names=[]
self.setLoggerLevel(level)
if server:
self.setupLogger(level)
def initInterface(self,testbed):
self.testbed=testbed
userdatadir = get_user_data_dir()
if not os.path.isdir(userdatadir):
os.mkdir(userdatadir)
if not os.path.isdir(os.path.join(userdatadir, "log")):
os.mkdir(os.path.join(userdatadir, "log"))
dbname= os.path.join(get_user_data_dir(), "log",'vizir.db')
self.dbpool=adbapi.ConnectionPool('sqlite3',dbname,check_same_thread=False)
self.id=0
self.lastId=0
d=self.deleteDB()
d.addCallback(self.createDB)
def init_config(filename=None):
global config, configfile,chConfig,chConfigFile
if filename == None:
dirname=get_user_data_dir()
if not os.path.isdir(dirname):
os.mkdir(dirname)
chFilename=os.path.join(dirname, "chConfig.cfg")
chConfigFile = chFilename
if not os.path.exists(chFilename):
ch=open(chFilename,'wb')
ch.close()
chConfig=ConfigParser.ConfigParser()
filename = os.path.join(dirname, "config.cfg")
if not os.path.exists(filename):
create_default_config(filename)
configfile = filename
return (configfile,chConfigFile)
chConfig = ConfigParser.ConfigParser()
config = ConfigParser.ConfigParser()
if config.read(filename) == []:
print "Unable to load configuration file at [%s] or file not found" % (filename)
raise ValueError(filename, 'Config file not found or unparsable')
else:
configfile = filename
try:
chConfig.read(chConfigFile)
except:
pass
return (configfile,chConfigFile)
def start(self, port='Server'):
userdatadir = get_user_data_dir()
if not os.path.isdir(userdatadir):
os.mkdir(userdatadir)
if not os.path.isdir(os.path.join(userdatadir, "log")):
os.mkdir(os.path.join(userdatadir, "log"))
n = 'p2ner' + str(port) + '.db'
dbname = os.path.join(get_user_data_dir(), "log", n)
print dbname
self.dbpool = adbapi.ConnectionPool('sqlite3',
dbname,
check_same_thread=False)
d = self.deleteDB()
d.addCallback(self.createDB)
d.addCallback(self.setReady)
class interface(object):
specs = {'statsDir': os.path.join(get_user_data_dir(), 'stats')}
specsGui = {
'statsDir': {
'name': 'Statistics Directory',
'tooltip': 'The directory to save statistics',
'type': 'browseDir'
}
}
def initStats(self, *args, **kwargs):
if kwargs.has_key('statsDir'):
self.dir=kwargs['statsDir']
else:
self.dir=get_user_data_dir()
self.statkeys = {}
self.fds = {}
self.loop = task.LoopingCall(self.writeLog)
self.loop.start(1.0)
self.lpb = -1
def getRemoteFileStat(self,filename):
if not filename:
return
baseFile=os.path.basename(filename)
# targetFile=os.path.join(get_user_data_dir(),'stats')
targetFile=get_user_data_dir()
targetFile=os.path.join(targetFile,'r'+baseFile)
d=self.parent.interface.copyStatFile(filename)
d.addCallback(self.writeRemoteStatFile,targetFile)
def on_loadTButton_clicked(self,widget):
for sub in widget.get_children():
widget.remove(sub)
directory=os.path.join(get_user_data_dir(),'stats')
files=[f[:-5] for f in os.listdir(directory) if '.tmpl' in f]
for f in files:
l=gtk.MenuItem(f)
l.show()
l.connect('activate',self.loadFile,f)
widget.append(l)
def loadFile(self,widget,filename):
if self.statinGraph:
print "can't load templeta while making grph"
return
filename+='.tmpl'
directory=os.path.join(get_user_data_dir(),'stats')
filename=os.path.join(directory,filename)
f=open(filename,'rb')
graph=load(f)
f.close()
self.makeTemplateGraph(graph)
def getParameters(self):
self.visibleCols = {}
self.showNetAtStart = config.getCheckNetMessages()
try:
self.convertedDir = config.config.get('General', 'cdir')
except:
dir = get_user_data_dir()
self.convertedDir = dir
config.config.set('General', 'cdir', dir)
try:
self.checkAtStart = config.config.getboolean(
'General', 'checkAtStart')
except:
self.checkAtStart = False
config.config.set('General', 'checkAtStart', 'false')
try:
self.subEncoding = config.config.get('General', 'subsencoding')
except:
self.subEncoding = 'Greek'
config.config.set('General', 'subsencoding', 'Greek')
self.tempSubEncoding = self.subEncoding
self.servers = config.get_servers()
try:
self.defaultServer = config.config.get('DefaultServ', 'ip')
except:
self.defaultServer = None
try:
self.upnp = config.config.getboolean('UPNP', 'on')
except:
self.upnp = True
config.config.set('UPNP', 'on', 'true')
self.remotePrefs = config.getRemotePreferences()
self.channels = config.get_channels()
if self.func:
self.func()
def initPlugin(self, *args, **kwargs):
if kwargs.has_key('statsDir'):
dir = kwargs['statsDir']
else:
dir = get_user_data_dir()
if not os.path.isdir(dir):
os.mkdir(dir)
self.peers = {}
self.systemStats = []
self.systemPeers = []
self.lpb = -1
self.db = database(dir)
self.subscribers = {}
self.loop = task.LoopingCall(self.updateDatabase)
self.loop.start(2.0)
self.waitingStats = {}
self.waitingStatsCount = 0
def initStats(self, *args, **kwargs):
if kwargs.has_key('statsDir'):
dir = kwargs['statsDir']
else:
dir = get_user_data_dir()
port = kwargs['port']
if not os.path.isdir(dir):
os.mkdir(dir)
self.statkeys = {}
self.lpb = -1
self.startTime = None
self.db = database(dir, port)
self.subscribers = {}
self.loop = task.LoopingCall(self.updateDatabase)
self.loop.start(2.0)
self.waitingStats = {}
self.waitingStatsCount = 0
def create_remote_config(file,chfile,remote):
global config, configfile, chConfig,chConfigFile
dirname=get_user_data_dir()
if not os.path.isdir(dirname):
os.mkdir(dirname)
if remote:
name="rconfig.cfg"
n='rChConfig.cfg'
else:
name="config.cfg"
n='chConfig.cfg'
filename = os.path.join(dirname, name)
f=open(filename,'wb')
for line in file:
f.write(line)
f.close()
filename2 = os.path.join(dirname, n)
f=open(filename2,'wb')
if chfile!=-1:
for line in chfile:
f.write(line)
f.close()
config = ConfigParser.ConfigParser()
if config.read(filename) == []:
print "Unable to load configuration file at [%s] or file not found" % (filename)
raise ValueError(filename, 'Config file not found or unparsable')
else:
configfile = filename
chConfig=ConfigParser.ConfigParser()
try:
chConfig.read(filename2)
except:
pass
chConfigFile=filename2
return (configfile,chConfigFile)
def on_sButton_clicked(self, widget):
en = self.builder.get_object('fileEntry').get_text()
if not en:
return
file = os.path.join(get_user_data_dir(), 'stats')
if not os.path.isdir(file):
os.mkdir(file)
en1 = en + '.stat'
f = open(os.path.join(file, en1), 'w')
for s, v in self.stats.items():
f.write(s + '\n')
en2 = en + s
f2 = open(os.path.join(file, en2), 'w')
for i in range(len(v['x'])):
line = str(v['x'][i]) + ',' + str(
v['y'][i] / v['scale']) + '\n'
f2.write(line)
f2.close()
f.close()
self.on_cButton_clicked()
def browseRemote(self):
RemoteFileChooser(self.getRemoteFileStat,self.parent.interface,dname=get_user_data_dir())
