AhmadYarAI commited on
Commit
cd9bdbe
·
1 Parent(s): 5e7b698

Done the configuration

Browse files
Files changed (1) hide show
  1. api/invoices.py +4 -3
api/invoices.py CHANGED
@@ -4,13 +4,14 @@ from datetime import datetime
4
  from typing import Optional, List
5
  from pydantic import BaseModel
6
 
 
7
  try:
8
- from backend.models.invoice import Invoice
9
  from backend.core.database import get_db
10
  except ImportError:
11
- from models.invoice import Invoice
12
  from core.database import get_db
13
-
14
  router = APIRouter(prefix="/invoices", tags=["Invoices"])
15
 
16
  class ItemCreate(BaseModel):
 
4
  from typing import Optional, List
5
  from pydantic import BaseModel
6
 
7
+ # SMART IMPORT for both environments
8
  try:
9
+ from backend.models.invoice import Invoice, InvoiceItem # Add InvoiceItem here
10
  from backend.core.database import get_db
11
  except ImportError:
12
+ from models.invoice import Invoice, InvoiceItem # And add it here
13
  from core.database import get_db
14
+
15
  router = APIRouter(prefix="/invoices", tags=["Invoices"])
16
 
17
  class ItemCreate(BaseModel):