2 Commits

Author SHA1 Message Date
laktionov-as 3099949f6c Написали класс для сканирования 2026-06-10 19:40:24 +03:00
laktionov-as b7d3c30c86 добавил main.py 2026-06-10 18:40:40 +03:00
+75 -59
View File
@@ -1,63 +1,79 @@
import requests
from bs4 import BeautifulSoup
import re
import sqlite3
from datetime import datetime
db_name = 'mydb.db'
class Spyder:
def __init__(self):
self.db_name = 'database.db'
self.conn = sqlite3.connect(self.db_name)
self.cursor = self.conn.cursor()
def init_db(self):
query = '''CREATE TABLE links(
id INTEGER PRIMARY KEY AUTOINCREMENT,
link TEXT NOT NULL,
last_scan INTEGER DEFAULT 0,
title TEXT DEFAULT ''
)'''
self.cursor.execute(query)
def scan(self,url):
if url.endswith('.pdf'):
return
headers = {
'User-Agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/148.0.0.0 Safari/537.36'
}
try:
response = requests.get(
url,
headers=headers,
timeout=5
)
except:
self.update_url(url,'Scan error')
return
html = response.text
soup = BeautifulSoup(html,'html.parser')
result = soup.find_all("a", href=True)
title = soup.title.text
self.update_url(url,title)
for el in result:
if el['href'].endswith('.pdf'):
continue
if not el['href'].startswith('http'):
match = re.findall(r'https?://[A-Za-z0-9.-]+',url)
link = f"{match[0]}{el['href']}"
else:
link = el['href']
query = f'SELECT id FROM links WHERE link = ?'
self.cursor.execute(query,(link,))
if self.cursor.fetchone():
continue
query = f'INSERT INTO links(link) VALUES(?)'
self.cursor.execute(query,(link,))
self.conn.commit()
def update_url(self,url,title):
query = "SELECT id FROM links WHERE link = ?"
self.cursor.execute(query,(url,))
link_id = self.cursor.fetchone()
if link_id:
query = "UPDATE links SET title=?, last_scan=? WHERE id=?"
self.cursor.execute(query,(title,datetime.now().timestamp(),link_id[0]))
self.conn.commit()
def get_url(self):
query = "SELECT link FROM links ORDER BY last_scan LIMIT 1"
self.cursor.execute(query)
return self.cursor.fetchone()
def close(self):
self.conn.close()
class User:
def __init__(self,name,email,age):
self.__id = None
self.__name = name
self.__email = email
self.__age = age
if __name__ == '__main__':
spyder = Spyder()
#spyder.init_db()
#spyder.scan('https://habr.com/ru/news/1046177/')
for _ in range(5):
url = spyder.get_url()
print(url[0])
spyder.scan(url[0])
spyder.close()
@property
def name(self):
return self.__name
@name.setter
def name(self,value):
self.__name = value
@property
def email(self):
return self.__email
@email.setter
def email(self,value):
if '@' in value:
self.__email = value
else:
raise ValueError('Не корректный адрес почты')
@property
def age(self):
return self.__age
@age.setter
def age(self,value):
if 0 <= value <= 120:
self.__age = value
else:
raise ValueError('Неправильно указан возраст')
def save(self):
conn = sqlite3.connect(db_name)
cursor = conn.cursor()
property_list = [f"{k}" for k, v in User.__dict__.items() if isinstance(v, property)]
if '_User__id' in self.__dict__.keys():
update_string = ','.join([f'`{p_name}`= ? ' for p_name in property_list])
update_values = [getattr(self,p) for p in property_list] + [self.__id]
query = f"UPDATE {User.__name__} SET {update_string} WHERE id = ?"
print(query,update_values)
else:
property_count = ('?,'*len(property_list))[:-1] #','.join(['?' for _ in range(len(property_list))])
property_names = ','.join(property_list)
property_values = [getattr(self,p) for p in property_list]
query = f"INSERT INTO {User.__name__}({property_names}) VALUES({property_count})"
print(query,property_values)
cursor.execute(query,property_values)
conn.commit()
self.__id = cursor.lastrowid
conn.close()
#print(User.__dict__)
user1 = User('Igor','igor@example.com',44)
user1.save()
user1.age = 55
user1.save()