def get_conference_sessions(wsck, typeOfSession=None):
conference = ndb.Key(urlsafe=wsck)
sessions = Session.query(ancestor=conference)
if typeOfSession is not None:
sessions = sessions.filter(Session.typeOfSession == typeOfSession)
return sessions
def set_featured_speaker(wsck, speaker_name):
speaker = get_create_speaker(speaker_name)
query = Session.query()
query = query.filter(Session.speaker == speaker)
sessions = query.filter(Session.websafeConferenceKey == wsck)
if sessions.count() > 1:
message = MESSAGE_FEATURED_SPEAKER.format(
speaker_name,
", ".join([sess.name for sess in sessions]))
memcache.set(MEMCACHE_FEATURED_SPEAKER, message)
def __translate_unscanned(self, value=None):
"""
Add subquerie as filter to keep only services that have not been
scanned yet (i.e. with no checks already run)
"""
session = Session()
return not_(
session.query(Result) \
.filter(Service.id == Result.service_id) \
.exists()
)
def __init__(self):
try:
print(BANNER)
# Parse settings files
settings = Settings()
# Parse command-line arguments
arguments = ArgumentsParser(settings)
# Create db if needed and initialize sqlalchemy session
Base.metadata.create_all(engine)
session = Session()
# Create "default" mission if necessary
mission = session.query(Mission).filter(
Mission.name == 'default').first()
if not mission:
mission = Mission(name='default', comment='Default scope')
session.add(mission)
session.commit()
# Controller
controller = MainController(arguments, settings, session)
controller.run()
except KeyboardInterrupt:
print()
if Output.prompt_confirm('Are you sure you want to exit ?',
default=False):
logger.error('User aborted')
sys.exit(0)
except (SettingsException, AttackException) as e:
logger.error(e)
sys.exit(1)
except (ValueError, ArgumentsException):
print
sys.exit(1)
except Exception as e:
print
logger.error('Unexpected error occured: {0}'.format(str(e)))
traceback.print_exc()
sys.exit(1)
def query_session_task3(last_hour, session_type):
query = Session.query()
query = query.filter(Session.startTime.IN(range(1, last_hour+1)))
query = query.filter(Session.typeOfSession != session_type)
return query
def get_sessions_by_speaker(speaker_name):
speaker = get_create_speaker(speaker_name)
query = Session.query()
query = query.filter(Session.speaker == speaker)
return query
def get_sessions():
query = Session.query()
return query.fetch()
