Python Store.add_record Examples

Example #1

File: manage.py Project: vibhusharma17/flask-rest-api

def load_data(test=False):
    with open("StoreMasterAssignment.csv") as f:
        datalist = [{k: str(v)
                     for k, v in row.items()}
                    for row in csv.DictReader(f, skipinitialspace=True)]

    if test is True:
        datalist = datalist[:5]

    for d in datalist:
        d["store_id"] = d.pop("StoreID")
        d["store_name"] = d.pop("StoreName")
        d["area_code"] = d.pop("AreaCode")
        Store.add_record(**d)
    print("store created", len(datalist))

Example #2

File: test_model.py Project: vibhusharma17/flask-rest-api

 def test_add_record(self):
     data = {"store_id": "XYZ123", "store_name": "XYZ", "area_code": "123"}
     Store.add_record(**data)
     row = Store.find_one(store_id="XYZ123")
     self.assertNotEqual(None, row)