def create_home_dir():
'''
创建home目录。
@return: 成功True。
'''
homeDir = get_home_dir()
if not os.path.exists(homeDir):
try:
os.makedirs(homeDir)
except OSError:
return False
# profile目录
profilesPath = homeDir + os.path.sep + 'profiles'
defaultProfilesPath = get_local_dir() + os.path.sep + 'profiles' + os.path.sep
if not os.path.exists(profilesPath):
try:
shutil.copytree(defaultProfilesPath, profilesPath)
except OSError:
return False
return True
self._write_to_file(div_end_str)
div_end_str = u"\t\t</div>\n" u"\t</div>\n"
self._write_to_file(div_end_str)
html_end_str = u"</div>\n</body>\n</html>\n"
self._write_to_file(html_end_str)
except Exception, e:
logger.error("Why this happen, report will return none:%s" % e)
self._file.close()
return None
else:
self._file.close()
datetimestrf = datetime.now().strftime("%Y-%m-%d")
import shutil
typestr = "恶意内容扫描报告"
final_report_path = get_local_dir() + os.path.sep + "reports" + os.path.sep
final_name = self.target + "_" + datetimestrf + "_" + typestr + ".html"
fin_path = os.path.join(final_report_path, final_name)
if not os.path.exists(final_report_path):
os.mkdir(final_report_path)
shutil.move(self._file_path, fin_path)
return fin_path
div_end_str = u'\t\t</div>\n' \
u'\t</div>\n'
self._write_to_file(div_end_str)
html_end_str = u'</div>\n</body>\n</html>\n'
self._write_to_file(html_end_str)
except Exception, e:
logger.error('Why this happen, report will return none:%s' % e)
self._file.close()
return None
else:
self._file.close()
datetimestrf = datetime.now().strftime('%Y-%m-%d')
import shutil
typestr = '恶意内容扫描报告'
final_report_path = get_local_dir(
) + os.path.sep + 'reports' + os.path.sep
final_name = self.target + '_' + datetimestrf + '_' + typestr + '.html'
fin_path = os.path.join(final_report_path, final_name)
if not os.path.exists(final_report_path):
os.mkdir(final_report_path)
shutil.move(self._file_path, fin_path)
return fin_path
return None
def getProfileValue(self, sectionName, optionName, valueType='string'):
'''
获取配置文件指定section、option的值。
@param sectionName: 配置区域名称。
@param optionName: 配置选项名称。
@param valueType: 返回值类型，取值‘int’、‘string’、'list’
'''
for section in self._config.sections():
if section == sectionName:
for option in self._config.options(section):
if option == optionName:
if valueType == 'string':
return self._config.get(section, option)
elif valueType == 'int':
return self._config.getint(section, option)
else:
aList = []
for value in self._config.get(section,
option).split(','):
aList.append(value)
return aList
return None
pf = Profile(get_local_dir() + os.path.sep + 'conf.dark')
return None
def getProfileValue(self, sectionName, optionName, valueType='string'):
'''
获取配置文件指定section、option的值。
@param sectionName: 配置区域名称。
@param optionName: 配置选项名称。
@param valueType: 返回值类型，取值‘int’、‘string’、'list’
'''
for section in self._config.sections():
if section == sectionName:
for option in self._config.options(section):
if option == optionName:
if valueType == 'string':
return self._config.get(section, option)
elif valueType == 'int':
return self._config.getint(section, option)
else:
aList = []
for value in self._config.get(section, option).split(','):
aList.append(value)
return aList
return None
pf = Profile(get_local_dir() + os.path.sep + 'conf.dark')
