/usr/lib/python2.7/dist-packages/pytrainer/profile.py is in pytrainer 1.10.2~git20160107-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 | # -*- coding: iso-8859-1 -*-
#Copyright (C) Fiz Vazquez vud1@sindominio.net
# Modified by dgranda
#This program is free software; you can redistribute it and/or
#modify it under the terms of the GNU General Public License
#as published by the Free Software Foundation; either version 2
#of the License, or (at your option) any later version.
#This program is distributed in the hope that it will be useful,
#but WITHOUT ANY WARRANTY; without even the implied warranty of
#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
#GNU General Public License for more details.
#You should have received a copy of the GNU General Public License
#along with this program; if not, write to the Free Software
#Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
import os, stat
import logging
from StringIO import StringIO
from lxml import etree
from lib.ddbb import DDBB
from lib.uc import UC
class Profile:
def __init__(self, environment, data_path = None, parent = None):
logging.debug(">>")
self.environment = environment
self.pytrainer_main = parent
self.data_path = data_path
self.xml_tree = None
self.tmpdir = environment.temp_dir
self.confdir = environment.conf_dir
self.conffile = environment.conf_file
self.gpxdir = environment.gpx_dir
self.extensiondir = environment.extension_dir
self.plugindir = environment.plugin_dir
self.uc = UC()
self.profilewindow = None
environment.clear_temp_dir()
#Profile Options and Defaults
self.profile_options = {
"prf_name":"default",
"prf_gender":"",
"prf_weight":"",
"prf_height":"",
"prf_age":"",
"prf_ddbb":"sqlite",
"prf_ddbbhost":"",
"prf_ddbbname":"",
"prf_ddbbuser":"",
"prf_ddbbpass":"",
"version":"0.0",
"prf_us_system":"False",
"prf_hrzones_karvonen":"False",
"prf_maxhr":"",
"prf_minhr":"",
"auto_launch_file_selection":"False",
"import_default_tab":"0",
"default_viewer":"0",
"window_size":"800, 640",
"activitypool_size": "10",
}
#Parse pytrainer configuration file
self.config_file = environment.conf_file
self.configuration = self._parse_config_file(self.config_file)
logging.debug("Configuration retrieved: "+str(self.configuration))
#self.pytrainer_main.ddbb = DDBB(self, pytrainer_main=self.pytrainer_main)
self.uc.set_us(self.prf_us_system)
self._setZones()
logging.debug("<<")
def _setZones(self):
#maxhr = self.getValue("pytraining","prf_maxhr")
#resthr = self.getValue("pytraining","prf_minhr")
try:
maxhr = int(self.getValue("pytraining","prf_maxhr"))
except Exception as e:
logging.error("Failed when retrieving Max Heartrate value: %s" %e)
logging.info("Setting Max Heartrate to default value: 190")
maxhr = 190
try:
resthr = int(self.getValue("pytraining","prf_minhr"))
except Exception as e:
logging.error("Failed when retrieving Min Heartrate value: %s" %e)
logging.info("Setting Min Heartrate to default value: 65")
resthr = 65
self.maxhr = maxhr
self.rethr = resthr
if self.getValue("pytraining","prf_hrzones_karvonen")=="True":
#karvonen method
targethr1 = ((maxhr - resthr) * 0.50) + resthr
targethr2 = ((maxhr - resthr) * 0.60) + resthr
targethr3 = ((maxhr - resthr) * 0.70) + resthr
targethr4 = ((maxhr - resthr) * 0.80) + resthr
targethr5 = ((maxhr - resthr) * 0.90) + resthr
targethr6 = maxhr
else:
#not karvonen method
targethr1 = maxhr * 0.50
targethr2 = maxhr * 0.60
targethr3 = maxhr * 0.70
targethr4 = maxhr * 0.80
targethr5 = maxhr * 0.90
targethr6 = maxhr
self.zone1 = (targethr1,targethr2,"#ffff99",_("Moderate activity"))
self.zone2 = (targethr2,targethr3,"#ffcc00",_("Weight Control"))
self.zone3 = (targethr3,targethr4,"#ff9900",_("Aerobic"))
self.zone4 = (targethr4,targethr5,"#ff6600",_("Anaerobic"))
self.zone5 = (targethr5,targethr6,"#ff0000",_("VO2 MAX"))
def getMaxHR(self):
return self.maxhr
def getRestHR(self):
return self.resthr
def getZones(self):
return self.zone5,self.zone4,self.zone3,self.zone2,self.zone1
def getConfFile(self):
if not os.path.isfile(self.conffile):
return False
else:
return self.conffile
def _parse_config_file(self, config_file):
'''
Parse the xml configuration file and convert to a dict
returns: dict with option as key
'''
if config_file is None:
logging.error("Configuration file value not set")
logging.error("Fatal error, exiting")
exit(-3)
if not os.path.isfile(config_file): #File not found
logging.error("Configuration '%s' file does not exist" % config_file)
logging.info("No profile found. Creating default one")
self.setProfile(self.profile_options)
if os.stat(config_file)[stat.ST_SIZE] == 0: #File is empty
logging.error("Configuration '%s' file is empty" % config_file)
logging.info("Creating default profile")
self.setProfile(self.profile_options)
logging.debug("Attempting to parse content from "+ config_file)
try:
parser = etree.XMLParser(encoding='UTF8', recover=True)
self.xml_tree = etree.parse(config_file, parser=parser)
#Have a populated xml tree, get pytraining node (root) and convert it to a dict
pytraining_tag = self.xml_tree.getroot()
config = {}
config_needs_update = False
for key, default in self.profile_options.items():
value = pytraining_tag.get(key)
#If property is not found, set it to the default
if value is None:
config_needs_update = True
value = default
config[key] = value
#Added a property, so update config
if config_needs_update:
self.setProfile(config)
#Set shorthand var for units of measurement
self.prf_us_system = True if config["prf_us_system"] == "True" else False
return config
except Exception as e:
logging.error("Error parsing file: %s. Exiting" % config_file)
logging.error(str(e))
logging.error("Fatal error, exiting")
exit(-3)
def getIntValue(self, tag, variable, default=0):
''' Function to return conf value as int
returns
-- default if cannot convert to int
-- None if variable not found
'''
result = self.getValue(tag, variable)
if result is None:
return None
try:
result = int(result)
except Exception as e:
logging.debug(str(e))
result = default
return result
def getValue(self, tag, variable):
if tag != "pytraining":
print "ERROR - pytraining is the only profile tag supported"
return None
elif not self.configuration.has_key(variable):
return None
return self.configuration[variable]
def setValue(self, tag, variable, value, delay_write=False):
logging.debug(">>")
if tag != "pytraining":
print "ERROR - pytraining is the only profile tag supported"
logging.debug("Setting %s to %s" % (variable, value))
if self.xml_tree is None:
#new config file....
self.xml_tree = etree.parse(StringIO('''<?xml version='1.0' encoding='UTF-8'?><pytraining />'''))
self.xml_tree.getroot().set(variable, value.decode('utf-8'))
if not delay_write:
logging.debug("Writting...")
self.xml_tree.write(self.config_file, xml_declaration=True, encoding='UTF-8')
logging.debug("<<")
def setProfile(self,list_options):
logging.debug(">>")
for option, value in list_options.items():
logging.debug("Adding "+option+"|"+value)
self.setValue("pytraining",option,value,delay_write=True)
self.xml_tree.write(self.config_file, xml_declaration=True, encoding='UTF-8')
self.uc.set_us(list_options['prf_us_system'])
logging.debug("<<")
def editProfile(self, sport_service):
logging.debug(">>")
from gui.windowprofile import WindowProfile
logging.debug("retrieving configuration data")
#Refresh configuration
self.configuration = self._parse_config_file(self.config_file)
if self.profilewindow is None:
self.profilewindow = WindowProfile(sport_service, self.data_path, self, pytrainer_main=self.pytrainer_main)
logging.debug("setting data values")
self.profilewindow.setValues(self.configuration)
self.profilewindow.run()
self.profilewindow = None
else:
self.profilewindow.setValues(self.configuration)
self.profilewindow.present()
self.configuration = self._parse_config_file(self.config_file)
logging.debug("<<")
def actualize_mainsportlist(self):
logging.debug("--")
self.pytrainer_main.refreshMainSportList()
|