biosample.management.commands.patch_orphan_biosamples.Command.handle() - Code Metrics - Inspection of ":sparkles: search and track orphaned BioSamples ID..." - cnr-ibba/IMAGE-InjectTool - Measure and Improve Code Quality continuously with Scrutinizer

Completed

Pull Request — devel (#90)

by Paolo

created 2020-05-30 15:25 UTC

Command.handle() B

↳ Parent: biosample.management.commands.patch_orphan_biosamples

Complexity

Conditions

Size

Total Lines	63
Code Lines	35

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
eloc	35
dl	0
loc	63
rs	8.5733
c	0
b	0
f	0
cc	5
nop	3

How to fix Long Method

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Tue May 26 17:21:28 2020

@author: Paolo Cozzi <[email protected]>
"""

import logging

from pyUSIrest.usi import Root
from pyUSIrest.exceptions import USIDataError

from django.core.management import BaseCommand

from biosample.helpers import get_manager_auth
from biosample.models import OrphanSubmission
from biosample.tasks.cleanup import get_orphan_samples
from common.constants import SUBMITTED, ERROR

# Get an instance of a logger
logger = logging.getLogger(__name__)


def create_biosample_submission(root, team):
    logger.debug("Creating a new submission")

    usi_team = root.get_team_by_name(team.name)
    usi_submission = usi_team.create_submission()

    submission = OrphanSubmission(
        usi_submission_name=usi_submission.name)
    submission.save()

    logger.debug("Created submission %s" % submission)
    return usi_submission, submission


def update_submission_status(submission):
    """Set SUBMITTED status to submission"""

    # update submission status
    if submission:
        submission.status = SUBMITTED
        submission.save()


class Command(BaseCommand):
    help = 'Get a JSON for biosample submission'

    def handle(self, *args, **options):
        # call commands and fill tables.
        logger.info("Called patch_orphan_samples")

        # get a new auth object
        auth = get_manager_auth()

        # get a new root object
        root = Root(auth)

        # some variables
        count = 1
        old_team = None
        usi_submission = None
        submission = None

        # iterate among orphan sample, create a BioSample submission
        # and then add biosample for patch
        for record in get_orphan_samples():
            (data, team, sample) = (
                record['data'], record['team'], record['sample'])

            if count % 100 == 0 or old_team != team:
                update_submission_status(submission)

                # create a new Biosample submission
                usi_submission, submission = create_biosample_submission(
                    root, team)

                # reset count and old_team
                count = 1
                old_team = team

            # add sample to submission
            try:
                logger.info("Submitting '%s'" % data['accession'])
                usi_submission.create_sample(data)

            except USIDataError as error:
                logger.error(
                    "Can't remove '%s': Error was '%s'" % (
                        data['accession'], str(error)))
                sample.status = ERROR
                sample.save()
                continue

            # update sample status and track submission
            sample.status = SUBMITTED
            sample.submission = submission
            sample.save()

            # update submission count
            submission.samples_count = count
            submission.save()

            # new element
            count += 1

        # update the last submission status
        update_submission_status(submission)

        # end the script
        logger.info("patch_orphan_samples ended")


1			#!/usr/bin/env python3
2			# -- coding: utf-8 --
3			"""
4			Created on Tue May 26 17:21:28 2020
5
6			@author: Paolo Cozzi <[email protected]>
7			"""
8
9			import logging
10
11			from pyUSIrest.usi import Root
12			from pyUSIrest.exceptions import USIDataError
13
14			from django.core.management import BaseCommand
15
16			from biosample.helpers import get_manager_auth
17			from biosample.models import OrphanSubmission
18			from biosample.tasks.cleanup import get_orphan_samples
19			from common.constants import SUBMITTED, ERROR
20
21			# Get an instance of a logger
22			logger = logging.getLogger(__name__)
23
24
25			def create_biosample_submission(root, team):
26			logger.debug("Creating a new submission")
27
28			usi_team = root.get_team_by_name(team.name)
29			usi_submission = usi_team.create_submission()
30
31			submission = OrphanSubmission(
32			usi_submission_name=usi_submission.name)
33			submission.save()
34
35			logger.debug("Created submission %s" % submission)
36			return usi_submission, submission
37
38
39			def update_submission_status(submission):
40			"""Set SUBMITTED status to submission"""
41
42			# update submission status
43			if submission:
44			submission.status = SUBMITTED
45			submission.save()
46
47
48			class Command(BaseCommand):
49			help = 'Get a JSON for biosample submission'
50
51			def handle(self, args, *options):
52			# call commands and fill tables.
53			logger.info("Called patch_orphan_samples")
54
55			# get a new auth object
56			auth = get_manager_auth()
57
58			# get a new root object
59			root = Root(auth)
60
61			# some variables
62			count = 1
63			old_team = None
64			usi_submission = None
65			submission = None
66
67			# iterate among orphan sample, create a BioSample submission
68			# and then add biosample for patch
69			for record in get_orphan_samples():
70			(data, team, sample) = (
71			record['data'], record['team'], record['sample'])
72
73			if count % 100 == 0 or old_team != team:
74			update_submission_status(submission)
75
76			# create a new Biosample submission
77			usi_submission, submission = create_biosample_submission(
78			root, team)
79
80			# reset count and old_team
81			count = 1
82			old_team = team
83
84			# add sample to submission
85			try:
86			logger.info("Submitting '%s'" % data['accession'])
87			usi_submission.create_sample(data)
88
89			except USIDataError as error:
90			logger.error(
91			"Can't remove '%s': Error was '%s'" % (
92			data['accession'], str(error)))
93			sample.status = ERROR
94			sample.save()
95			continue
96
97			# update sample status and track submission
98			sample.status = SUBMITTED
99			sample.submission = submission
100			sample.save()
101
102			# update submission count
103			submission.samples_count = count
104			submission.save()
105
106			# new element
107			count += 1
108
109			# update the last submission status
110			update_submission_status(submission)
111
112			# end the script
113			logger.info("patch_orphan_samples ended")
114

cnr-ibba / IMAGE-InjectTool

Pull Request — devel (#90)

Command.handle() B

Complexity

Size

Duplication

Importance

How to fix Long Method

Long Method

Duplication Side-by-Side

Filter issues like