From 2949e52b829a3eede9664cc4a0b84a776bddceaa Mon Sep 17 00:00:00 2001
From: Mike Glenn <mglenn@ilude.com>
Date: Sat, 15 Jun 2024 10:59:21 -0400
Subject: [PATCH] remove pyping

---
 app/processors/title_editor.py | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/app/processors/title_editor.py b/app/processors/title_editor.py
index e9ea76e..ff1fc55 100644
--- a/app/processors/title_editor.py
+++ b/app/processors/title_editor.py
@@ -1,7 +1,7 @@
 from datetime import datetime, timedelta
 import logging
 import os
-import pyping
+import socket
 
 from pytz import utc
 from models.feed_article import FeedArticle
@@ -18,12 +18,17 @@
 
 
 class TitleEditor:
+  def hostname_resolves(hostname):
+    try:
+        socket.gethostbyname(hostname)
+        return 1
+    except socket.error:
+        return 0
+    
   def __init__(self):
     self.ollama_url = os.getenv('OLLAMA_URL')
 
-    self.ping_result = (pyping.ping(self.ollama_url).ret_code == 0)
-
-    if self.ollama_url and self.ping_result:
+    if self.ollama_url and self.hostname_resolves(self.ollama_url):
       parser = StructuredOutputParser.from_response_schemas(
           [ResponseSchema(name="title", description="title of the article")]
       )
@@ -72,7 +77,7 @@ def __init__(self):
       self.script_hash = calculate_sha1_hash(f"{system_prompt.content}{model_name}{model_temp}")
 
   def process(self, articles: list[FeedArticle]) -> list[FeedArticle]:
-    if self.ollama_url and self.ping_result:
+    if self.ollama_url and self.hostname_resolves(self.ollama_url):
 
       needs_processed = list(filter(lambda article: article.processed != self.script_hash, articles))
       # if len(needs_processed) > 10: