mangroveorg/datawinners

import logging
from datawinners.main.database import get_db_manager
from datawinners.project.view_models import ReporterEntity
from datawinners.tasks import app
from mangrove.datastore.entity import get_by_short_code, contact_by_short_code
from mangrove.form_model.form_model import REPORTER
from mangrove.transport.contract.survey_response import SurveyResponse


@app.task(max_retries=3, throw=False)
def update_datasender_on_open_submissions(database_name, reporter_id):
    logger = logging.getLogger('datawinners.tasks')
    try:
        dbm = get_db_manager(database_name)
        logger.error(reporter_id)

        reporter_entity = ReporterEntity(contact_by_short_code(dbm, reporter_id))
        rows = dbm.load_all_rows_in_view("anonymous_submissions", key=reporter_entity.mobile_number)

        for row in rows:
            _update_survey_response(dbm, row, reporter_entity.entity.id)

    except Exception as e:
        logger.exception('Failed for db: %s ,reporter_id: %s' % (database_name, reporter_id))
        logger.exception(e)


def _update_survey_response(dbm, row, reporter_id):
    survey_response_doc = SurveyResponse.__document_class__.wrap(row['value'])
    survey_response = SurveyResponse.new_from_doc(dbm=dbm, doc=survey_response_doc)
    survey_response.is_anonymous_submission = False
    survey_response.owner_uid = reporter_id
    survey_response.save()

@app.task(max_retries=3, throw=False)
def convert_open_submissions_to_registered_submissions(database_name, reporter_ids):
        for reporter_id in reporter_ids:
            update_datasender_on_open_submissions(database_name, reporter_id)