import galaxy.model
from galaxy.model.orm import *
from galaxy.model.mapping import context as sa_session
from base.twilltestcase import *
import sys

def delete_obj( obj ):
    sa_session.delete( obj )
    sa_session.flush()
def delete_request_type_permissions( id ):
    rtps = sa_session.query( galaxy.model.RequestTypePermissions ) \
                     .filter( and_( galaxy.model.RequestTypePermissions.table.c.request_type_id==id ) ) \
                     .order_by( desc( galaxy.model.RequestTypePermissions.table.c.create_time ) )
    for rtp in rtps:
        sa_session.delete( rtp )
    sa_session.flush()
def delete_user_roles( user ):
    for ura in user.roles:
        sa_session.delete( ura )
    sa_session.flush()
def flush( obj ):
    sa_session.add( obj )
    sa_session.flush()
def get_all_histories_for_user( user ):
    return sa_session.query( galaxy.model.History ) \
                     .filter( and_( galaxy.model.History.table.c.user_id==user.id,
                                    galaxy.model.History.table.c.deleted==False ) ) \
                     .all()
def get_dataset_permissions_by_dataset( dataset ):
    return sa_session.query( galaxy.model.DatasetPermissions ) \
                     .filter( galaxy.model.DatasetPermissions.table.c.dataset_id==dataset.id ) \
                     .all()
def get_dataset_permissions_by_role( role ):
    return sa_session.query( galaxy.model.DatasetPermissions ) \
                     .filter( galaxy.model.DatasetPermissions.table.c.role_id == role.id ) \
                     .first()
def get_default_history_permissions_by_history( history ):
    return sa_session.query( galaxy.model.DefaultHistoryPermissions ) \
                     .filter( galaxy.model.DefaultHistoryPermissions.table.c.history_id==history.id ) \
                     .all()
def get_default_history_permissions_by_role( role ):
    return sa_session.query( galaxy.model.DefaultHistoryPermissions ) \
                     .filter( galaxy.model.DefaultHistoryPermissions.table.c.role_id == role.id ) \
                     .all()
def get_default_user_permissions_by_role( role ):
    return sa_session.query( galaxy.model.DefaultUserPermissions ) \
                     .filter( galaxy.model.DefaultUserPermissions.table.c.role_id == role.id ) \
                     .all()
def get_default_user_permissions_by_user( user ):
    return sa_session.query( galaxy.model.DefaultUserPermissions ) \
                     .filter( galaxy.model.DefaultUserPermissions.table.c.user_id==user.id ) \
                     .all()
def get_form( name ):
    fdc_list = sa_session.query( galaxy.model.FormDefinitionCurrent ) \
                         .filter( galaxy.model.FormDefinitionCurrent.table.c.deleted == False ) \
                         .order_by( galaxy.model.FormDefinitionCurrent.table.c.create_time.desc() )
    for fdc in fdc_list:
        sa_session.refresh( fdc )
        sa_session.refresh( fdc.latest_form )
        if fdc.latest_form.name == name:
            return fdc.latest_form
    return None
def get_folder( parent_id, name, description ):
    return sa_session.query( galaxy.model.LibraryFolder ) \
                     .filter( and_( galaxy.model.LibraryFolder.table.c.parent_id==parent_id,
                                    galaxy.model.LibraryFolder.table.c.name==name,
                                    galaxy.model.LibraryFolder.table.c.description==description ) ) \
                     .first()
def get_group_by_name( name ):
    return sa_session.query( galaxy.model.Group ).filter( galaxy.model.Group.table.c.name==name ).first()
def get_group_role_associations_by_group( group ):
    return sa_session.query( galaxy.model.GroupRoleAssociation ) \
                     .filter( galaxy.model.GroupRoleAssociation.table.c.group_id == group.id ) \
                     .all()
def get_group_role_associations_by_role( role ):
    return sa_session.query( galaxy.model.GroupRoleAssociation ) \
                     .filter( galaxy.model.GroupRoleAssociation.table.c.role_id == role.id ) \
                     .all()
def get_latest_dataset():
    return sa_session.query( galaxy.model.Dataset ) \
                     .order_by( desc( galaxy.model.Dataset.table.c.create_time ) ) \
                     .first()
def get_latest_hda():
    return sa_session.query( galaxy.model.HistoryDatasetAssociation ) \
                     .order_by( desc( galaxy.model.HistoryDatasetAssociation.table.c.create_time ) ) \
                     .first()
def get_latest_history_for_user( user ):
    return sa_session.query( galaxy.model.History ) \
                     .filter( and_( galaxy.model.History.table.c.deleted==False,
                                    galaxy.model.History.table.c.user_id==user.id ) ) \
                     .order_by( desc( galaxy.model.History.table.c.create_time ) ) \
                     .first()
def get_latest_ldda_by_name( name ):
    return sa_session.query( galaxy.model.LibraryDatasetDatasetAssociation ) \
                     .filter( and_( galaxy.model.LibraryDatasetDatasetAssociation.table.c.name==name,
                                    galaxy.model.LibraryDatasetDatasetAssociation.table.c.deleted == False ) ) \
                     .order_by( desc( galaxy.model.LibraryDatasetDatasetAssociation.table.c.create_time ) ) \
                     .first()
def get_latest_lddas( limit ):
    return sa_session.query( galaxy.model.LibraryDatasetDatasetAssociation ) \
                     .order_by( desc( galaxy.model.LibraryDatasetDatasetAssociation.table.c.update_time ) ) \
                     .limit( limit )
def get_library( name, description, synopsis ):
    return sa_session.query( galaxy.model.Library ) \
                     .filter( and_( galaxy.model.Library.table.c.name==name,
                                    galaxy.model.Library.table.c.description==description,
                                    galaxy.model.Library.table.c.synopsis==synopsis,
                                    galaxy.model.Library.table.c.deleted==False ) ) \
                     .first()
def get_private_role( user ):
    for role in user.all_roles():
        if role.name == user.email and role.description == 'Private Role for %s' % user.email:
            return role
    raise AssertionError( "Private role not found for user '%s'" % user.email )
def get_request_by_name( name ):
    return sa_session.query( galaxy.model.Request ) \
                     .filter( and_( galaxy.model.Request.table.c.name==name,
                                    galaxy.model.Request.table.c.deleted==False ) ) \
                     .first()   
def get_request_type_by_name( name ):
    return sa_session.query( galaxy.model.RequestType ) \
                     .filter( and_( galaxy.model.RequestType.table.c.name==name ) ) \
                     .order_by( desc( galaxy.model.RequestType.table.c.create_time ) ) \
                     .first()
def get_role_by_name( name ):
    return sa_session.query( galaxy.model.Role ).filter( galaxy.model.Role.table.c.name==name ).first()
def get_user( email ):
    return sa_session.query( galaxy.model.User ) \
                     .filter( galaxy.model.User.table.c.email==email ) \
                     .first()
def get_user_address( user, short_desc ):
    return sa_session.query( galaxy.model.UserAddress ) \
                     .filter( and_( galaxy.model.UserAddress.table.c.user_id==user.id,
                                    galaxy.model.UserAddress.table.c.desc==short_desc,
                                    galaxy.model.UserAddress.table.c.deleted==False ) ) \
                     .order_by( desc( galaxy.model.UserAddress.table.c.create_time ) ) \
                     .first()   
def get_user_group_associations_by_group( group ):
    return sa_session.query( galaxy.model.UserGroupAssociation ) \
                     .filter( galaxy.model.UserGroupAssociation.table.c.group_id == group.id ) \
                     .all()
def get_user_info_form_definition():
    return galaxy.model.FormDefinition.types.USER_INFO
def get_user_role_associations_by_role( role ):
    return sa_session.query( galaxy.model.UserRoleAssociation ) \
                     .filter( galaxy.model.UserRoleAssociation.table.c.role_id == role.id ) \
                     .all()
def mark_obj_deleted( obj ):
    obj.deleted = True
    sa_session.add( obj )
    sa_session.flush()
def refresh( obj ):
    sa_session.refresh( obj )