@@ -1005,14 +1005,14 @@ def check_study_name(self, study_name):
10051005 """ Verifies the provided study name is valid and not already in use. """
10061006 if not study_name .replace ('_' , '' ).isalnum ():
10071007 raise StudyNameError ("Only alpha numeric characters and '_' are allowed in the study name" )
1008- if MMEDSDoc .objects (study_name = study_name , doc_type = 'study' ):
1008+ if MMEDSDoc .objects (study_name = study_name , doc_type = DocType . STUDY ):
10091009 raise StudyNameError (f"Study name { study_name } already in use" )
10101010
10111011 def check_sequencing_run_name (self , run_name ):
10121012 """ Verifies the provided sequencing run name is valid and not already in use. """
10131013 if not run_name .replace ('_' , '' ).isalnum ():
10141014 raise StudyNameError ("Only alpha-numeric characters and '_' are allowed in the sequencing run name" )
1015- if MMEDSDoc .objects (study_name = run_name , doc_type = 'sequencing_run' ):
1015+ if MMEDSDoc .objects (data_name = run_name , doc_type = DocType . DATA ):
10161016 raise StudyNameError (f"Sequencing Run name { run_name } already in use" )
10171017
10181018 def get_sequencing_run_locations (self , metadata , user , column = ("RawDataProtocol" , "RawDataProtocolID" )):
@@ -1042,27 +1042,27 @@ def get_sequencing_run_locations(self, metadata, user, column=("RawDataProtocol"
10421042
10431043 def get_all_studies (self ):
10441044 """ Return all studies currently stored in the database. """
1045- return MMEDSDoc .objects (doc_type = 'study' )
1045+ return MMEDSDoc .objects (doc_type = DocType . STUDY )
10461046
10471047 def get_all_analyses (self ):
10481048 """ Return all analyses currently stored in the database. """
1049- return MMEDSDoc .objects (doc_type = 'analysis' )
1049+ return MMEDSDoc .objects (doc_type = DocType . ANALYSIS )
10501050
10511051 def get_all_sequencing_runs (self ):
10521052 """ Return all sequencing runs currently stored in the database. """
1053- return MMEDSDoc .objects (doc_type = 'sequencing_run' )
1053+ return MMEDSDoc .objects (doc_type = DocType . DATA )
10541054
10551055 def get_all_user_sequencing_runs (self , user ):
10561056 """ Return all sequencing runs currently stored in the database owned by USER. """
1057- return MMEDSDoc .objects (doc_type = 'sequencing_run' , owner = user )
1057+ return MMEDSDoc .objects (doc_type = DocType . DATA , owner = user )
10581058
10591059 def get_all_user_studies (self , user ):
10601060 """ Return all studies currently stored in the database owned by USER. """
1061- return MMEDSDoc .objects (doc_type = 'study' , owner = user )
1061+ return MMEDSDoc .objects (doc_type = DocType . STUDY , owner = user )
10621062
10631063 def get_all_analyses_from_study (self , access_code ):
10641064 """ Return all studies currently stored in the database. """
1065- return MMEDSDoc .objects (study_code = access_code , doc_type = 'analysis' )
1065+ return MMEDSDoc .objects (study_code = access_code , doc_type = DocType . ANALYSIS )
10661066
10671067 def check_files (self , access_code ):
10681068 """ Check that all files associated with the study actually exist. """
@@ -1076,7 +1076,7 @@ def check_files(self, access_code):
10761076
10771077 def get_metadata_file_location (self , study_name ):
10781078 """ Return the metadata.tsv file location for a given study """
1079- doc = MMEDSDoc .objects (doc_type = 'study' , study_name = study_name ).first ()
1079+ doc = MMEDSDoc .objects (doc_type = DocType . STUDY , study_name = study_name ).first ()
10801080 return doc ['files' ]['metadata' ]
10811081
10821082 def delete_mongo_documents (self ):
0 commit comments