예제 #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()
예제 #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()
예제 #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)
예제 #4
0
파일: base.py 프로젝트: d-kahara/hairq-API
    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()
예제 #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)
예제 #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()
예제 #7
0
파일: base.py 프로젝트: KMaina/S.C.M
    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()
예제 #8
0
파일: base.py 프로젝트: marcdomain/mrm_api
    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()
예제 #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()
예제 #10
0
 def save(self):
     """Function for saving new objects"""
     db_session.add(self)
     db_session.commit()
예제 #11
0
 def delete(self):
     """Function for deleting objects"""
     db_session.delete(self)
     db_session.commit()
예제 #12
0
파일: base.py 프로젝트: verenceLola/mrm_api
    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')
예제 #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)
예제 #14
0
 def save(self):
     """Function that saves new objects"""
     db_session.add(self)
     db_session.commit()
예제 #15
0
 def delete(self):
     """Function that deletes objects"""
     db_session.delete(self)
     db_session.commit()