rafaldembski commited on
Commit
f18485b
verified
1 Parent(s): 1d76d81

Update utils/functions.py

Browse files
Files changed (1) hide show
  1. utils/functions.py +6 -3
utils/functions.py CHANGED
@@ -20,7 +20,8 @@ logging.basicConfig(
20
  )
21
 
22
  # Konfiguracja bazy danych SQLite
23
- DATABASE_URL = "sqlite:///scam_detector.db"
 
24
 
25
  engine = create_engine(DATABASE_URL, connect_args={"check_same_thread": False})
26
  SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
@@ -60,6 +61,7 @@ def add_fake_number(phone_number):
60
  """
61
  db = SessionLocal()
62
  try:
 
63
  existing_number = db.query(FakeNumber).filter(FakeNumber.phone_number == phone_number).first()
64
  if not existing_number:
65
  new_number = FakeNumber(phone_number=phone_number)
@@ -188,8 +190,8 @@ def get_phone_info(phone_number):
188
  """
189
  try:
190
  parsed_number = phonenumbers.parse(phone_number, None)
191
- country = geocoder.description_for_number(parsed_number, 'pl')
192
- operator = carrier.name_for_number(parsed_number, 'pl')
193
  logging.info(f"Numer {phone_number} - Kraj: {country}, Operator: {operator}.")
194
  return country, operator
195
  except phonenumbers.NumberParseException as e:
@@ -355,3 +357,4 @@ Provide your analysis and conclusions following the guidelines above."""
355
  except Exception as e:
356
  logging.error(f"API Connection Error: {e}")
357
  return f"API Connection Error: {e}", "Analysis Error.", "Analysis Error."
 
 
20
  )
21
 
22
  # Konfiguracja bazy danych SQLite
23
+ BASE_DIR = os.path.dirname(os.path.abspath(__file__))
24
+ DATABASE_URL = f"sqlite:///{os.path.join(BASE_DIR, '..', 'scam_detector.db')}"
25
 
26
  engine = create_engine(DATABASE_URL, connect_args={"check_same_thread": False})
27
  SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 
61
  """
62
  db = SessionLocal()
63
  try:
64
+ logging.info(f"Pr贸buj臋 doda膰 numer: {phone_number}")
65
  existing_number = db.query(FakeNumber).filter(FakeNumber.phone_number == phone_number).first()
66
  if not existing_number:
67
  new_number = FakeNumber(phone_number=phone_number)
 
190
  """
191
  try:
192
  parsed_number = phonenumbers.parse(phone_number, None)
193
+ country = geocoder.description_for_number(parsed_number, 'en') # Mo偶esz zmieni膰 j臋zyk na 'pl' lub 'de'
194
+ operator = carrier.name_for_number(parsed_number, 'en') # Mo偶esz zmieni膰 j臋zyk na 'pl' lub 'de'
195
  logging.info(f"Numer {phone_number} - Kraj: {country}, Operator: {operator}.")
196
  return country, operator
197
  except phonenumbers.NumberParseException as e:
 
357
  except Exception as e:
358
  logging.error(f"API Connection Error: {e}")
359
  return f"API Connection Error: {e}", "Analysis Error.", "Analysis Error."
360
+