Beispiel #1
0
    def add_node(self, **kwargs):
        """Function for adding nodes"""
        db_session.add(
            OfficeStructure(name=kwargs['name'],
                            parent_id=kwargs.get('parent_id', None),
                            tag_id=kwargs.get('tag_id', None)))

        db_session.commit()
Beispiel #2
0
    def add_branch(self, nodes):
        """
        Function to add a single branch
        :params list of nodes
        """
        db_session.add_all(nodes)

        db_session.commit()
Beispiel #3
0
 def mutate(self, info, node_id):
     exact_node = db_session.query(StructureModel).filter(
         StructureModel.id == node_id).first()
     if not exact_node:
         raise GraphQLError("The specified node does not exist")
     db_session.delete(exact_node)
     db_session.commit()
     return DeleteNode(node=exact_node)
Beispiel #4
0
    def setUp(self):
        app = self.create_app()
        self.app_test = app.test_client()
        with app.app_context():
            Base.metadata.create_all(bind=engine)

            command.stamp(self.alembic_configuration, 'head')
            command.downgrade(self.alembic_configuration, '-1')
            command.upgrade(self.alembic_configuration, 'head')
            db_session.commit()
Beispiel #5
0
 def mutate(self, info, node_list):
     validate_structure_nodes(node_list)
     admin_location_id = admin_roles.user_location_for_analytics_view()
     nodes = []
     for node in node_list:
         validate_empty_fields(**node)
         node['name'] = node.name.strip()
         node['tag'] = node.tag.strip()
         nodes.append(StructureModel(**node, location_id=admin_location_id))
     db_session.add_all(nodes)
     db_session.commit()
     return CreateStructure(structure=nodes)
Beispiel #6
0
 def setUp(self):
     app = self.create_app()
     self.app_test = app.test_client()
     with app.app_context():
         Base.metadata.create_all(bind=engine)
         admin_user = User(email="*****@*****.**",
                           location="Kampala",
                           name="Peter Walugembe",
                           picture="https://www.andela.com/walugembe")
         admin_user.save()
         role = Role(role="Admin")
         role.save()
         user_role = UsersRole(user_id=admin_user.id, role_id=role.id)
         user_role.save()
         location = Location(name='Kampala', abbreviation='KLA')
         location.save()
         location_two = Location(name='Nairobi', abbreviation='NBO')
         location_two.save()
         office = Office(name="St. Catherines", location_id=location.id)
         office.save()
         office_two = Office(name="dojo", location_id=location_two.id)
         office_two.save()
         block = Block(name='EC', office_id=office.id)
         block.save()
         floor = Floor(name='3rd', block_id=block.id)
         floor.save()
         room = Room(
             name='Entebbe',
             room_type='meeting',
             capacity=6,
             floor_id=floor.id,
             calendar_id=
             '*****@*****.**',  # noqa: E501
             image_url=
             "https://www.officelovin.com/wp-content/uploads/2016/10/andela-office-main-1.jpg"
         )  # noqa: E501
         room.save()
         resource = Resource(name='Markers', quantity=3, room_id=room.id)
         resource.save()
         device = Devices(resource_id=resource.id,
                          last_seen="2018-06-08T11:17:58.785136",
                          date_added="2018-06-08T11:17:58.785136",
                          name="Samsung",
                          location="Nairobi",
                          device_type="External Display")
         device.save()
         db_session.commit()
Beispiel #7
0
    def setUp(self):
        app = self.create_app()
        self.app_test = app.test_client()

        with app.app_context():
            Base.metadata.create_all(bind=engine)
            command.stamp(self.alembic_configuration, 'head')
            # command.downgrade(self.alembic_configuration, '-1')
            command.upgrade(self.alembic_configuration, 'head')

            user = User(name="Ken", password="******", user_type='default')
            user.save()
            product = Product(name="Eggs",
                              manufacturer="none",
                              price=200,
                              tax=0,
                              UOM="none",
                              user_id="1")
            product.save()
            db_session.commit()
Beispiel #8
0
    def setUp(self):
        app = self.create_app()
        self.app_test = app.test_client()
        with app.app_context():
            Base.metadata.create_all(bind=engine)

            command.stamp(self.alembic_configuration, 'head')
            command.downgrade(self.alembic_configuration, '-1')
            command.upgrade(self.alembic_configuration, 'head')

            admin_user = User(email="*****@*****.**",
                              location="Kampala",
                              name="Peter Walugembe",
                              picture="https://www.andela.com/walugembe")
            admin_user.save()
            lagos_admin = User(email="*****@*****.**",
                               location="Lagos",
                               name="Peter Adeoye",
                               picture="https://www.andela.com/adeoye")
            lagos_admin.save()
            global role
            role = Role(role="Admin")
            role.save()
            admin_user.roles.append(role)
            lagos_admin.roles.append(role)
            tag = Tag(name='Block-B',
                      color='green',
                      description='The description')
            tag.save()
            root_node = OfficeStructure(name='location', tag_id=1)
            root_node.save()
            leaf_node = OfficeStructure(name='wings', parent_id=1)
            leaf_node.save()

            location = Location(name='Kampala',
                                abbreviation='KLA',
                                structure_id=1)
            location.save()
            location_two = Location(name='Nairobi',
                                    abbreviation='NBO',
                                    structure_id=1)
            location_two.save()
            location_three = Location(name='Lagos',
                                      abbreviation='LOS',
                                      structure_id=1)
            location_three.save()
            tag_two = Tag(name='Block-C',
                          color='blue',
                          description='The description')
            tag_two.save()
            room = Room(
                name='Entebbe',
                room_type='meeting',
                capacity=6,
                location_id=location.id,
                calendar_id=
                '*****@*****.**',  # noqa: E501
                image_url=
                "https://www.officelovin.com/wp-content/uploads/2016/10/andela-office-main-1.jpg"
            )  # noqa: E501
            room.save()
            room.room_tags.append(tag)
            resource = Resource(name='Markers', quantity=3, room_id=room.id)
            resource.save()
            device = Devices(last_seen="2018-06-08T11:17:58.785136",
                             date_added="2018-06-08T11:17:58.785136",
                             name="Samsung",
                             location="Nairobi",
                             device_type="External Display")
            device.save()
            question_1 = Question(
                question_type="rate",
                question_title="Rating Feedback",
                question="How will you rate the brightness of the room",
                start_date="20 Nov 2018",
                end_date="28 Nov 2018",
                is_active=True)
            question_1.save()
            question_2 = Question(
                question_type="check",
                question_title="check Feedback",
                question="Is there anything missing in the room",
                start_date="20 Nov 2018",
                end_date="30 Nov 2018",
                is_active=True)
            event = Events(event_id="test_id5",
                           room_id=1,
                           event_title="Onboarding",
                           start_time="2018-07-11T09:00:00Z",
                           end_time="2018-07-11T09:45:00Z",
                           number_of_participants=4,
                           checked_in=False,
                           cancelled=False)
            event.save()
            question_2.save()
            question_3 = Question(question_type="input",
                                  question_title="input Feedback",
                                  question="Any other suggestion",
                                  start_date="20 Nov 2018",
                                  end_date="28 Nov 2018")
            question_3.save()
            response_1 = Response(
                question_id=1,
                room_id=1,
                rate=2,
                created_date=datetime.now(),
                resolved=False,
            )
            response_1.save()
            response_2 = Response(
                question_id=question_2.id,
                room_id=room.id,
                check=True,
                created_date=datetime.now(),
                resolved=True,
            )
            response_2.save()
            response_2.missing_resources.append(resource)
            structure = Structure(
                structure_id='b05fc5f2-b4aa-4f48-a8fb-30bdcc3fc968',
                level=1,
                name='Epic tower',
                parent_id="1",
                parent_title="parent_title",
                tag='Building',
                location_id=1,
                position=1,
            )
            structure.save()
            db_session.commit()
Beispiel #9
0
    def setUp(self):
        app = self.create_app()
        self.app_test = app.test_client()
        with app.app_context():
            Base.metadata.create_all(bind=engine)
            admin_user = User(first_name="doe",
                              last_name="jon",
                              other_names="smith",
                              email="*****@*****.**",
                              password="******",
                              picture="https://picsum.photos/200",
                              user_type="admin")

            admin_user.save()
            citizen_user = User(first_name="doe",
                                last_name="jon",
                                other_names="smith",
                                email="*****@*****.**",
                                password="******",
                                picture="https://picsum.photos/200",
                                user_type="citizen")

            citizen_user.save()
            politician_user = User(first_name="doe",
                                   last_name="jon",
                                   other_names="smith",
                                   email="*****@*****.**",
                                   password="******",
                                   picture="https://picsum.photos/200",
                                   user_type="politician")

            politician_user.save()
            dup_politician_user = User(first_name="doe",
                                       last_name="jon",
                                       other_names="smith",
                                       email="*****@*****.**",
                                       password="******",
                                       picture="https://picsum.photos/200",
                                       user_type="politician")

            dup_politician_user.save()
            party = Party(
                party_name="party",
                hq_address=
                "5 City Of Power Avenue, Somolu, Lagos, Nigeria",  # noqa
                logo_url="www.ipsum/pic")
            party.save()
            party_2 = Party(
                party_name="second_party",
                hq_address=
                "5 City Of Power Avenue, Somolu, Lagos, Nigeria",  # noqa
                logo_url="www.ipsum/pic")
            party_2.save()
            office = Office(office_name="office",
                            office_type="state",
                            age_limit=50,
                            description="my testing office")
            office.save()
            candidate = Candidate(user_id=4, office_id=1, party_id=1)
            candidate.save()
            db_session.commit()
Beispiel #10
0
 def save(self):
     """Function for saving new objects"""
     db_session.add(self)
     db_session.commit()
Beispiel #11
0
 def delete(self):
     """Function for deleting objects"""
     db_session.delete(self)
     db_session.commit()
Beispiel #12
0
    def setUp(self, mock_verify_calendar_id):
        app = self.create_app()
        self.app_test = app.test_client()
        with app.app_context():
            Base.metadata.create_all(bind=engine)

            command.stamp(self.alembic_configuration, 'head')
            command.downgrade(self.alembic_configuration, '-1')
            command.upgrade(self.alembic_configuration, 'head')

            admin_user = User(email="*****@*****.**",
                              name="Peter Walugembe",
                              picture="https://www.andela.com/walugembe")
            admin_user.location = "Kampala"
            admin_user.save()
            lagos_admin = User(email="*****@*****.**",
                               location="Lagos",
                               name="Peter Adeoye",
                               picture="https://www.andela.com/adeoye")
            lagos_admin.save()
            global role
            role = Role(role="Admin")
            role.save()
            role_2 = Role(role="Test")
            role_2.save()
            role_3 = Role(role="Super Admin")
            role_3.save()
            admin_user.roles.append(role)
            lagos_admin.roles.append(role)
            tag = Tag(name='Block-B',
                      color='green',
                      description='The description')
            tag.save()

            location = Location(name='Kampala', abbreviation='KLA')
            location.save()
            location_two = Location(name='Nairobi', abbreviation='NBO')
            location_two.save()
            location_three = Location(name='Lagos', abbreviation='LOS')
            location_three.save()
            tag_two = Tag(name='Block-C',
                          color='blue',
                          description='The description')
            tag_two.save()
            room = Room(
                name='Entebbe',
                room_type='meeting',
                capacity=6,
                location_id=location.id,
                structure_id='851ae8b3-48dd-46b5-89bc-ca3f8111ad87',
                calendar_id=
                '*****@*****.**',  # noqa: E501
                image_url=
                "https://www.officelovin.com/wp-content/uploads/2016/10/andela-office-main-1.jpg",  # noqa: E501
                room_labels=["1st Floor", "Wing A"])
            room.save()
            room.room_tags.append(tag)
            room_2 = Room(
                name='Tana',
                room_type='meeting',
                capacity=14,
                location_id=location.id,
                structure_id='851ae8b3-48dd-46b5-89bc-ca3f8111ad87',
                calendar_id=
                '*****@*****.**',  # noqa: E501
                image_url=
                "https://www.officelovin.com/wp-content/uploads/2016/10/andela-office-main-1.jpg",  # noqa: E501
                room_labels=["1st Floor", "Wing B"])
            room_2.save()
            room_2.room_tags.append(tag)
            resource = Resource(name='Markers', quantity=3)
            resource.save()
            device = Devices(last_seen="2018-06-08T11:17:58.785136",
                             date_added="2018-06-08T11:17:58.785136",
                             name="Samsung",
                             location="Kampala",
                             device_type="External Display",
                             room_id=1,
                             state="active")
            device.save()
            question_1 = Question(
                question_type="rate",
                question_title="Rating Feedback",
                question="How will you rate the brightness of the room",
                start_date="20 Nov 2018",
                end_date="28 Nov 2018",
                is_active=True)
            question_1.save()
            question_2 = Question(
                question_type="check",
                question_title="check Feedback",
                question="Is there anything missing in the room",
                check_options=['apple tv', 'whiteboard', 'maker pen'],
                start_date="20 Nov 2018",
                end_date="30 Nov 2018",
                is_active=True)
            event = Events(event_id="test_id5",
                           room_id=1,
                           event_title="Onboarding",
                           start_time="2018-07-11T09:00:00Z",
                           end_time="2018-07-11T09:45:00Z",
                           number_of_participants=4,
                           checked_in=False,
                           cancelled=False)
            event.save()
            question_2.save()
            question_3 = Question(question_type="input",
                                  question_title="input Feedback",
                                  question="Any other suggestion",
                                  start_date="20 Nov 2018",
                                  end_date="28 Nov 2018")
            question_3.save()
            question_4 = Question(question_type="check",
                                  question_title="Missing item",
                                  question="Anything missing in the room?",
                                  check_options=['duster'],
                                  start_date="20 Nov 2018",
                                  end_date="30 Nov 2018",
                                  is_active=True)
            question_4.save()
            response_1 = Response(
                question_id=1,
                room_id=1,
                question_type="rate",
                created_date=datetime.now(),
                response="1",
                resolved=False,
            )
            response_1.save()

            response_2 = Response(
                question_id=question_2.id,
                room_id=room.id,
                question_type="check",
                created_date=datetime.now(),
                response=['marker pen', 'apple tv'],
                resolved=True,
            )
            response_2.save()
            response_2.missing_resources.append(resource)

            response_3 = Response(question_id=question_4.id,
                                  room_id=room_2.id,
                                  question_type="check",
                                  created_date=datetime.now(),
                                  response=['duster'],
                                  resolved=True,
                                  state="archived")
            response_3.save()

            structure = Structure(
                structure_id='b05fc5f2-b4aa-4f48-a8fb-30bdcc3fc968',
                level=1,
                name='Epic tower',
                parent_id="1",
                parent_title="parent_title",
                tag='Building',
                location_id=1,
                position=1,
            )
            structure.save()
            parent_node = OfficeStructure(
                id='C56A4180-65AA-42EC-A945-5FD21DEC0518',
                name='Epic Tower',
                tag='Lagos Building',
                location_id=1)
            parent_node.save()
            child_node = OfficeStructure(
                id='C56A4180-65AA-42EC-A945-5FD21DEC0519',
                name='Gold Coast',
                tag='First Floor',
                parent_id='C56A4180-65AA-42EC-A945-5FD21DEC0518',
                location_id=1)
            child_node.save()
            db_session.commit()
            f = open('mrm.err.log', 'a+')
            f.write('[2019-08-06 13:22:32 +0000] [1574] [ERROR] Error /logs\r')
            f.write('Traceback (most recent call last):\r')
            f.write('if pattern.search(line):\r')
Beispiel #13
0
    "post":
    "Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident."
}]

comments = [{
    "comment": "Such a goood idea"
}, {
    "comment": "So funny cant stop laughing"
}, {
    "comment": "Lol"
}]

for user in users:
    user_obj = User(**user)
    db_session.add(user_obj)
    db_session.commit()

for post in posts:
    post.update(
        {'user_id': random.choice([user.id for user in User.query.all()])})
    post_obj = Post(**post)
    db_session.add(post_obj)
    db_session.commit()
for comment in comments:
    print("Before updatte", comment)
    comment.update(
        {'post_id': random.choice([post.id for post in Post.query.all()])})
    comment.update(
        {'created_by': random.choice([user.id for user in User.query.all()])})
    print("After updatte", comment)
    comment_obj = Comment(**comment)
Beispiel #14
0
 def save(self):
     """Function that saves new objects"""
     db_session.add(self)
     db_session.commit()
Beispiel #15
0
 def delete(self):
     """Function that deletes objects"""
     db_session.delete(self)
     db_session.commit()