Commit 6c62c06b authored by arjun.b's avatar arjun.b

updated after review

(file upload and stored to database)
parents
URI="mongodb://localhost:27017"
# Default ignored files
/shelf/
/workspace.xml
<component name="InspectionProjectProfileManager">
<profile version="1.0">
<option name="myName" value="Project Default" />
<inspection_tool class="PyPep8NamingInspection" enabled="true" level="WEAK WARNING" enabled_by_default="true">
<option name="ignoredErrors">
<list>
<option value="N801" />
</list>
</option>
</inspection_tool>
</profile>
</component>
\ No newline at end of file
<component name="InspectionProjectProfileManager">
<settings>
<option name="USE_PROJECT_PROFILE" value="false" />
<version value="1.0" />
</settings>
</component>
\ No newline at end of file
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="ProjectRootManager" version="2" project-jdk-name="Python 3.9 (new_pymongo)" project-jdk-type="Python SDK" />
</project>
\ No newline at end of file
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="ProjectModuleManager">
<modules>
<module fileurl="file://$PROJECT_DIR$/.idea/new_pymongo.iml" filepath="$PROJECT_DIR$/.idea/new_pymongo.iml" />
</modules>
</component>
</project>
\ No newline at end of file
<?xml version="1.0" encoding="UTF-8"?>
<module type="PYTHON_MODULE" version="4">
<component name="NewModuleRootManager">
<content url="file://$MODULE_DIR$">
<excludeFolder url="file://$MODULE_DIR$/venv" />
</content>
<orderEntry type="jdk" jdkName="Python 3.9 (new_pymongo)" jdkType="Python SDK" />
<orderEntry type="sourceFolder" forTests="false" />
</component>
</module>
\ No newline at end of file
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="VcsDirectoryMappings">
<mapping directory="$PROJECT_DIR$" vcs="Git" />
</component>
</project>
\ No newline at end of file
import uvicorn
from scripts.config.application_config import server_path, port_no
if __name__ == "__main__":
try:
uvicorn.run(server_path, port=int(port_no), reload=True)
except Exception as e:
print(str(e))
[path]
base_path= scripts/temp/emp.csv
[mongo]
uri=mongodb://localhost:27017
port=27017
[server]
server_path=scripts.services.main:app
port_no=8000
import configparser
try:
config = configparser.ConfigParser()
config.read(r'E:\new_pymongo\conf\application.conf')
# path of csv file
file_path = config.get("path", "base_path")
# mongo connect
mongo_uri = config.get("mongo", "uri")
port_connect = config.get("mongo", "port")
# server
server_path = config.get("server", "server_path")
port_no = config.get("server", "port_no")
except configparser.NoOptionError:
print("could not read configuration file")
from pymongo import MongoClient
from scripts.config.application_config import mongo_uri
def db_connect():
try:
conn = MongoClient(mongo_uri)
database = conn.employees
# print(conn.list_database_names())
dblist = conn.list_database_names()
if database in dblist:
print("database exist")
print("mongo connect successfully")
return database
except Exception as e:
print(str(e))
from mongoengine import Document, StringField, ObjectIdField, IntField
from pydantic import BaseModel
class Employee(Document):
_id: ObjectIdField()
id: IntField()
first_name: StringField()
last_name: StringField()
gender: StringField()
phone: StringField()
class employee_insert(BaseModel):
_id: str
id: int
first_name: str
last_name: str
gender: str
phone: str
class employee_update(BaseModel):
id: int
class employee_delete(BaseModel):
id: int
import json
from scripts.services.main import collection
class fastAPIHandler:
def file_upload(self):
try:
data = json.loads(self.decode('utf-8'))
collection.insert_many(data)
except Exception as e:
print(e)
from fastapi import FastAPI, UploadFile
from mongoengine import connect
from scripts.config.application_config import port_connect
from scripts.constants.db_connection import db_connect
from scripts.core.database.model import employee_insert
from scripts.utils.mongo_utils import create_collection
from scripts.core.handlers.fastapi_handler import fastAPIHandler
app = FastAPI()
connect(db="employees", host="localhost", port=int(port_connect))
@app.get("/")
async def root():
return {"data": "FastAPI CRUD Operations"}
# upload and insert into database
@app.post("/upload", tags=["upload file"])
async def upload_file(file: UploadFile):
file_content = file.file.read()
await fastAPIHandler.file_upload(file_content)
return {"data": "CSV data uploaded successfully"}
# insert data using post method
@app.post("/send_data", tags=['send data'])
def send_data(emp: employee_insert):
new_employee = employee_insert(id=emp.id,
first_name=emp.first_name,
last_name=emp.last_name,
gender=emp.gender,
phone=emp.phone)
collection.insert_one(dict(new_employee))
return {"message": "new data has been inserted"}
# update data
@app.put("/update_data/{doc_id}", tags=["update database"])
def update_data(doc_id: int):
collection.update_many({"id": doc_id}, {"$set": {"first_name": "ALAN"}})
return {"data": "data updated"}
# delete data
@app.delete("/delete_data", tags=["delete data"])
def delete_data(doc_id: int):
collection.delete_many({"id": doc_id})
return {"data": "data deleted"}
# create database
db = db_connect()
print("database created")
# create collection
collection = create_collection(db)
print("collection created")
This diff is collapsed.
def create_collection(db):
try:
collection_name = "employee"
collection = db[collection_name]
return collection
except Exception as e:
print(str(e))
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment