def test_interests(self):
        if self.skip_tests:
            return

        interests = ["interest_number_{0}".format(i) for i in range(1, 9)]
        [InterestService.create(interest_name=i) for i in interests]
        interests = InterestService.get_all_interests().to_primitive(
        )["interests"]
        self.assertEqual(len(interests), 8)

        # Create new interest.
        new_interest = InterestService.create(
            interest_name="test_new_interest")
        interests = InterestService.get_all_interests().to_primitive(
        )["interests"]
        self.assertEqual(len(interests), 9)

        # Update interest
        new_interest.name = "new_interest_name"
        updated_interest = InterestService.update(new_interest.id,
                                                  new_interest)
        self.assertEqual(new_interest.name, updated_interest.name)

        # Associate users with interest.
        iids, uids = TestInterestService.relationship_user(
            interests, self.test_user.id, [1, 4, 5])
        self.assertEqual(uids, iids)

        # Try again.
        iids, uids = TestInterestService.relationship_user(
            interests, self.test_user.id, [1, 2])
        self.assertEqual(uids, iids)

        # Validate unexistent interest.
        with self.assertRaises(NotFound):
            InterestService.create_or_update_user_interests(
                self.test_user.id, [0])

        # Associate projects with interest.
        iids, pids = TestInterestService.relationship_project(
            interests, self.test_project.id, [1, 2])
        self.assertEqual(pids, iids)

        # Validate unexistent interest.
        with self.assertRaises(NotFound):
            InterestService.create_or_update_project_interests(
                self.test_project.id, [0])

        # Delete one by one.
        for interest in interests:
            InterestService.delete(interest["id"])

        interests = InterestService.get_all_interests().to_primitive(
        )["interests"]
        self.assertEqual(len(interests), 0)
Exemple #2
0
def update_project_categories(filename):
    with open(filename, "r", encoding="ISO-8859-1", newline="") as csvfile:
        reader = csv.DictReader(csvfile)
        for row in reader:
            project_id = int(row.get("projectId"))
            primary_category = row.get("primaryCat")
            interest_ids = []
            # Map only primary_category interest to projects
            if primary_category:
                try:
                    interest = InterestService.get_by_name(primary_category)
                except NotFound:
                    interest = InterestService.create(primary_category)
                interest_ids.append(interest.id)

            try:
                InterestService.create_or_update_project_interests(
                    project_id, interest_ids)
            except Exception as e:
                print(f"Problem updating {project_id}: {type(e)}")
Exemple #3
0
    def post(self):
        """
        Creates a new interest
        ---
        tags:
            - interests
        produces:
            - application/json
        parameters:
            - in: header
              name: Authorization
              description: Base64 encoded session token
              required: true
              type: string
              default: Token sessionTokenHere==
            - in: body
              name: body
              required: true
              description: JSON object for creating a new interest
              schema:
                  properties:
                      name:
                          type: string
                          default: Public Domain
        responses:
            200:
                description: New interest created
            400:
                description: Invalid Request
            401:
                description: Unauthorized - Invalid credentials
            403:
                description: Forbidden
            500:
                description: Internal Server Error
        """
        try:
            orgs_dto = OrganisationService.get_organisations_managed_by_user_as_dto(
                token_auth.current_user())
            if len(orgs_dto.organisations) < 1:
                raise ValueError("User not a Org Manager")
        except ValueError as e:
            error_msg = f"InterestsAllAPI POST: {str(e)}"
            return {"Error": error_msg}, 403

        try:
            interest_dto = InterestDTO(request.get_json())
            interest_dto.validate()
        except DataError as e:
            current_app.logger.error(f"Error validating request: {str(e)}")
            return str(e), 400

        try:
            new_interest = InterestService.create(interest_dto.name)
            return new_interest.to_primitive(), 200
        except IntegrityError:
            return (
                {
                    "error":
                    "Value '{0}' already exists".format(interest_dto.name)
                },
                400,
            )
        except Exception as e:
            error_msg = f"Interest POST - unhandled error: {str(e)}"
            current_app.logger.critical(error_msg)
            return {"Error": error_msg}, 500