.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "_examples/factory-boy/microbiology.py"
.. LINE NUMBERS ARE GIVEN BELOW.

.. only:: html

    .. note::
        :class: sphx-glr-download-link-note

        Click :ref:`here <sphx_glr_download__examples_factory-boy_microbiology.py>`
        to download the full example code

.. rst-class:: sphx-glr-example-title

.. _sphx_glr__examples_factory-boy_microbiology.py:


01. Generate microbiology data
------------------------------

.. GENERATED FROM PYTHON SOURCE LINES 6-130

.. code-block:: default
   :lineno-start: 8



    import factory
    import inspect

    # Specific
    from datetime import timedelta
    from random import randrange


    MICROORGANISMS = ['O%s'%i for i in range(10)]
    ANTIMICROBIALS = ['A%s'%i for i in range(10)]
    SPECIMENS = ['BLDCUL', 'URICUL']
    METHODS = ['DD']
    SENSITIVITIES = ['S', 'I', 'R']

    # -----------------------
    # Classes
    # -----------------------
    class Base:
        """"""
        def __init__(self, **attributes):
            for key, value in attributes.items():
                self.__dict__[key] = value

        def __str__(self):
            attrs = [i
                     for i in inspect.getmembers(self)
                     if not i[0].startswith('_') and
                     not inspect.ismethod(i[1])
            ]
            return ' | '.join(['%s:%s' % (k,v) for k,v in attrs])

    class Sample(Base):
        pass

    class Test(Base):
        pass


    # -----------------------
    # Factories
    # -----------------------
    class SampleFactory(factory.Factory):
        """Factory to create samples."""
        class Meta:
            model = Sample

        date_collected = factory.Faker('date_time')
        laboratory_number = factory.Sequence(lambda n: 'LAB%08d' % n)
        specimen = factory.Iterator(SPECIMENS)


    class TestFactory(factory.Factory):
        """Factory to create susceptibility tests."""
        sample = factory.SubFactory(SampleFactory)
        microorganism = factory.Iterator(MICROORGANISMS)
        antimicrobial = factory.Iterator(ANTIMICROBIALS)
        method = factory.Iterator(METHODS)
        sensitivity = factory.Iterator(SENSITIVITIES)

        @factory.lazy_attribute
        def date_received(self):
            return self.sample.date_collected + timedelta(hours=self.offset1)

        @factory.lazy_attribute
        def date_outcome(self):
            return self.date_received + timedelta(hours=self.offset2)

        class Meta:
            model = Test

        class Params:
            offset1 = randrange(3, 24)
            offset2 = randrange(30, 60)



        
    if __name__ == '__main__':

        # Libraries
        import random
        from random import randrange

        def create_sample_random(norgs, nabxs):
            """Creates samples (randomly)"""
            s = SampleFactory()
            for o in random.sample(MICROORGANISMS, norgs):
                for a in random.sample(ANTIMICROBIALS, nabxs):
                    t = TestFactory(sample=s, antimicrobial=a, microorganism=o)
                    print(t)

        def create_sample_seq(norgs, nabxs):
            """Creates samples (Iterator)

            .. note: Ensure that N_MAX_ORGS and N_MAX_ABXS are
                     lower than the lengths of the corresponding
                     items.
            """
            s = SampleFactory()
            for o in range(norgs):
                t = TestFactory.create_batch(nabxs, sample=s)
                for e in t:
                    print(e)


        # Constants
        N_SAMPLES = 2
        N_MAX_ORGS = 4
        N_MAX_ABXS = 6

        print("\nSEQUENTIAL:")
        for ni in range(N_SAMPLES):
           create_sample_seq(
               norgs=randrange(N_MAX_ORGS),
               nabxs=randrange(N_MAX_ABXS)
           )

        print("\nRANDOM:")
        for ni in range(N_SAMPLES):
           create_sample_random(
               norgs=randrange(N_MAX_ORGS),
               nabxs=randrange(N_MAX_ABXS)
           )

.. rst-class:: sphx-glr-timing

   **Total running time of the script:** ( 0 minutes  0.000 seconds)


.. _sphx_glr_download__examples_factory-boy_microbiology.py:


.. only :: html

 .. container:: sphx-glr-footer
    :class: sphx-glr-footer-example



  .. container:: sphx-glr-download sphx-glr-download-python

     :download:`Download Python source code: microbiology.py <microbiology.py>`



  .. container:: sphx-glr-download sphx-glr-download-jupyter

     :download:`Download Jupyter notebook: microbiology.ipynb <microbiology.ipynb>`


.. only:: html

 .. rst-class:: sphx-glr-signature

    `Gallery generated by Sphinx-Gallery <https://sphinx-gallery.github.io>`_