Commit 4150116e authored by Vasyl Bodnaruk's avatar Vasyl Bodnaruk

Refactor function for make companies

parent 7a0a722a
from scrapy.utils.project import get_project_settings
from ..helpers import CompanyMaker, Database
db_settings = get_project_settings().get('DB')
db = Database(**db_settings)
class QueryMixin:
def __init__(self, *args, **kwargs):
self.condition = kwargs.get('query')
......@@ -9,3 +16,8 @@ class QueryMixin:
def get_common_items(self, company):
return {'region_id': company.region_id, 'type_id': company.type_id,
'media_id': company.media_id, 'company_id': company.id}
def companies(self, name):
companies = CompanyMaker(db.select(self.query))
companies.make_companies(name)
return companies.get_companies()
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