Skip to content

Commit

Permalink
refactor: Use Client.session attribute where possible (#921)
Browse files Browse the repository at this point in the history
  • Loading branch information
edgarrmondragon authored Aug 2, 2023
1 parent f6115c4 commit 67f8c14
Showing 1 changed file with 36 additions and 36 deletions.
72 changes: 36 additions & 36 deletions src/citric/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -133,7 +133,7 @@ def __init__(

def close(self) -> None:
"""Close client session."""
self.__session.close()
self.session.close()

def __enter__(self: Self) -> Self:
"""Create client context."""
Expand Down Expand Up @@ -166,7 +166,7 @@ def get_fieldmap(self, survey_id: int) -> dict:
.. versionadded:: 0.3.0
"""
return self.__session.get_fieldmap(survey_id)
return self.session.get_fieldmap(survey_id)

def activate_survey(self, survey_id: int) -> types.OperationStatus:
"""Activate a survey.
Expand All @@ -181,7 +181,7 @@ def activate_survey(self, survey_id: int) -> types.OperationStatus:
.. versionadded:: 0.0.1
"""
return self.__session.activate_survey(survey_id)
return self.session.activate_survey(survey_id)

def activate_tokens(
self,
Expand All @@ -203,7 +203,7 @@ def activate_tokens(
.. versionadded:: 0.0.1
"""
return self.__session.activate_tokens(survey_id, attributes or [])
return self.session.activate_tokens(survey_id, attributes or [])

def add_language(self, survey_id: int, language: str) -> types.OperationStatus:
"""Add a survey language.
Expand All @@ -220,7 +220,7 @@ def add_language(self, survey_id: int, language: str) -> types.OperationStatus:
.. versionadded:: 0.0.10
"""
return self.__session.add_language(survey_id, language)
return self.session.add_language(survey_id, language)

def add_participants(
self,
Expand All @@ -245,7 +245,7 @@ def add_participants(
.. versionchanged:: 0.4.0
Use keyword-only arguments.
"""
return self.__session.add_participants(
return self.session.add_participants(
survey_id,
participant_data,
create_tokens,
Expand Down Expand Up @@ -320,7 +320,7 @@ def add_survey(
.. versionadded:: 0.0.10
"""
return self.__session.add_survey(
return self.session.add_survey(
survey_id,
title,
language,
Expand All @@ -345,7 +345,7 @@ def delete_participants(
.. versionadded:: 0.0.1
"""
return self.__session.delete_participants(
return self.session.delete_participants(
survey_id,
participant_ids,
)
Expand Down Expand Up @@ -422,7 +422,7 @@ def add_group(self, survey_id: int, title: str, description: str = "") -> int:
.. versionadded:: 0.0.8
"""
return self.__session.add_group(survey_id, title, description)
return self.session.add_group(survey_id, title, description)

def _add_response(
self,
Expand All @@ -439,7 +439,7 @@ def _add_response(
Returns:
ID of the new response.
"""
return int(self.__session.add_response(survey_id, response_data))
return int(self.session.add_response(survey_id, response_data))

def add_response(self, survey_id: int, response_data: t.Mapping[str, t.Any]) -> int:
"""Add a single response to a survey.
Expand Down Expand Up @@ -498,7 +498,7 @@ def update_response(self, survey_id: int, response_data: dict[str, t.Any]) -> bo
"""
questions = self._get_question_mapping(survey_id)
data = self._map_response_keys(response_data, questions)
return self.__session.update_response(survey_id, data)
return self.session.update_response(survey_id, data)

def copy_survey(self, survey_id: int, name: str) -> dict[str, t.Any]:
"""Copy a survey.
Expand All @@ -514,7 +514,7 @@ def copy_survey(self, survey_id: int, name: str) -> dict[str, t.Any]:
.. versionadded:: 0.0.10
"""
return self.__session.copy_survey(survey_id, name)
return self.session.copy_survey(survey_id, name)

def import_cpdb_participants(
self,
Expand Down Expand Up @@ -552,7 +552,7 @@ def delete_group(self, survey_id: int, group_id: int) -> int:
.. versionadded:: 0.0.10
"""
return self.__session.delete_group(survey_id, group_id)
return self.session.delete_group(survey_id, group_id)

def delete_language(self, survey_id: int, language: str) -> types.OperationStatus:
"""Delete a language from a survey.
Expand All @@ -567,7 +567,7 @@ def delete_language(self, survey_id: int, language: str) -> types.OperationStatu
.. versionadded:: 0.0.12
.. minlimesurvey:: 5.3.4
"""
return self.__session.delete_language(survey_id, language)
return self.session.delete_language(survey_id, language)

def delete_quota(self, quota_id: int) -> types.OperationStatus:
"""Delete a LimeSurvey quota.
Expand Down Expand Up @@ -601,7 +601,7 @@ def delete_response(
.. versionadded:: 0.0.2
"""
return self.__session.delete_response(survey_id, response_id)
return self.session.delete_response(survey_id, response_id)

def delete_question(self, question_id: int) -> int:
"""Delete a survey.
Expand All @@ -617,7 +617,7 @@ def delete_question(self, question_id: int) -> int:
.. versionadded:: 0.1.0
.. minlimesurvey:: 5.3.19
"""
return self.__session.delete_question(question_id)
return self.session.delete_question(question_id)

def delete_survey(self, survey_id: int) -> types.OperationStatus:
"""Delete a survey.
Expand All @@ -632,7 +632,7 @@ def delete_survey(self, survey_id: int) -> types.OperationStatus:
.. versionadded:: 0.0.1
"""
return self.__session.delete_survey(survey_id)
return self.session.delete_survey(survey_id)

def export_responses( # noqa: PLR0913
self,
Expand Down Expand Up @@ -674,7 +674,7 @@ def export_responses( # noqa: PLR0913
"""
if token is None:
return base64.b64decode(
self.__session.export_responses(
self.session.export_responses(
survey_id,
enums.ResponsesExportFormat(file_format),
language,
Expand All @@ -688,7 +688,7 @@ def export_responses( # noqa: PLR0913
)

return base64.b64decode(
self.__session.export_responses_by_token(
self.session.export_responses_by_token(
survey_id,
enums.ResponsesExportFormat(file_format),
token,
Expand Down Expand Up @@ -878,7 +878,7 @@ def get_group_properties(
.. versionadded:: 0.0.10
"""
return self.__session.get_group_properties(group_id, settings, language)
return self.session.get_group_properties(group_id, settings, language)

def get_language_properties(
self,
Expand All @@ -899,7 +899,7 @@ def get_language_properties(
.. versionadded:: 0.0.10
"""
return self.__session.get_language_properties(survey_id, settings, language)
return self.session.get_language_properties(survey_id, settings, language)

def get_participant_properties(
self,
Expand All @@ -922,7 +922,7 @@ def get_participant_properties(
.. versionadded:: 0.0.1
"""
return self.__session.get_participant_properties(survey_id, query, properties)
return self.session.get_participant_properties(survey_id, query, properties)

def get_question_properties(
self,
Expand All @@ -945,7 +945,7 @@ def get_question_properties(
.. versionadded:: 0.0.10
"""
return self.__session.get_question_properties(question_id, settings, language)
return self.session.get_question_properties(question_id, settings, language)

def get_quota_properties(
self,
Expand Down Expand Up @@ -988,7 +988,7 @@ def get_response_ids(
.. versionadded:: 0.0.1
"""
return self.__session.get_response_ids(survey_id, token)
return self.session.get_response_ids(survey_id, token)

def get_available_site_settings(self) -> list[str]:
"""Get all available site settings.
Expand Down Expand Up @@ -1016,7 +1016,7 @@ def _get_site_setting(self, setting_name: str) -> types.Result:
.. versionadded:: 0.0.1
"""
return self.__session.get_site_settings(setting_name)
return self.session.get_site_settings(setting_name)

def get_default_theme(self) -> str:
"""Get the global default theme.
Expand Down Expand Up @@ -1103,7 +1103,7 @@ def get_survey_properties(
.. versionadded:: 0.0.1
"""
return self.__session.get_survey_properties(survey_id, properties)
return self.session.get_survey_properties(survey_id, properties)

def get_uploaded_files(
self,
Expand All @@ -1123,7 +1123,7 @@ def get_uploaded_files(
.. versionadded:: 0.0.5
"""
return self.__session.get_uploaded_files(survey_id, token)
return self.session.get_uploaded_files(survey_id, token)

def get_uploaded_file_objects(
self,
Expand Down Expand Up @@ -1211,7 +1211,7 @@ def import_group(
.. versionadded:: 0.0.10
"""
contents = base64.b64encode(file.read()).decode()
return self.__session.import_group(
return self.session.import_group(
survey_id,
contents,
enums.ImportGroupType(file_type),
Expand Down Expand Up @@ -1242,7 +1242,7 @@ def import_question(
.. versionadded:: 0.0.8
"""
contents = base64.b64encode(file.read()).decode()
return self.__session.import_question(
return self.session.import_question(
survey_id,
group_id,
contents,
Expand Down Expand Up @@ -1277,7 +1277,7 @@ def import_survey(
Accept a binary file object instead of a path.
"""
contents = base64.b64encode(file.read()).decode()
return self.__session.import_survey(
return self.session.import_survey(
contents,
enums.ImportSurveyType(file_type),
survey_name,
Expand Down Expand Up @@ -1313,7 +1313,7 @@ def list_participants(
.. versionchanged:: 0.4.0
Use keyword-only arguments.
"""
return self.__session.list_participants(
return self.session.list_participants(
survey_id,
start,
limit,
Expand All @@ -1332,7 +1332,7 @@ def list_users(self) -> list[dict[str, t.Any]]:
.. versionadded:: 0.0.3
"""
return self.__session.list_users()
return self.session.list_users()

def list_groups(
self,
Expand All @@ -1352,7 +1352,7 @@ def list_groups(
.. versionadded:: 0.0.10
"""
return self.__session.list_groups(survey_id, language)
return self.session.list_groups(survey_id, language)

def list_questions(
self,
Expand All @@ -1374,7 +1374,7 @@ def list_questions(
.. versionadded:: 0.0.1
"""
return self.__session.list_questions(survey_id, group_id, language)
return self.session.list_questions(survey_id, group_id, language)

def list_quotas(self, survey_id: int) -> list[types.QuotaListElement]:
"""Get all quotas for a LimeSurvey survey.
Expand Down Expand Up @@ -1405,7 +1405,7 @@ def list_surveys(self, username: str | None = None) -> list[dict[str, t.Any]]:
.. versionadded:: 0.0.1
"""
return self.__session.list_surveys(username)
return self.session.list_surveys(username)

def list_survey_groups(
self,
Expand All @@ -1423,7 +1423,7 @@ def list_survey_groups(
.. versionadded:: 0.0.2
"""
return self.__session.list_survey_groups(username)
return self.session.list_survey_groups(username)

def set_group_properties(
self,
Expand Down

0 comments on commit 67f8c14

Please sign in to comment.