HW3

HW3.py

@@ -0,0 +1,263 @@
+import logging
+import re
+import psycopg2
+from log import logging_decor
+from conf import DATA_BASE, LOGIN, PASSWD, HST, NPORT
+
+conn = psycopg2.connect(database=DATA_BASE, user=LOGIN,
+    password=PASSWD, host=HST, port=NPORT)
+
+cur = conn.cursor()
+# cur.execute("DROP TABLE IF EXISTS patients")
+cur.execute("CREATE TABLE IF NOT EXISTS patients (patient_id SERIAL PRIMARY KEY,first_name VARCHAR(64), " + "last_name VARCHAR(64), birth_date VARCHAR(64), phone VARCHAR(64), " +"document_type VARCHAR(64), document_id VARCHAR(64))")
+conn.commit()
+
+
+
+class LogAndChange(object):
+    """Дескриптор данных, в котором организована проверка полей
+    из конструктора объекта Patient а также их изменение
+    и логирование(логгер определим вне классов)"""
+
+    # Проверка даты
+    def _check_date(self, val):
+        if len(val) == 10:  # Количество символов, включая знаки препинания и цифры, в верном формате равно 10
+            if not re.match('(\d{4})-(\d{2})-(\d{2})', val):  # проверка на НЕсоответсвие паттерну 1994-10-12
+                new_date = re.split('\.|/|-', val)  # Сформируем список из чисел по разделителям . / -
+                if len(new_date[2]) == 4:  # если год стоит на последнем месте
+                    new_date = new_date[::-1]  # исправим это
+                new_string = '-'.join(new_date)  # соберем все вместе в правильно формате 1994-10-12
+                val = new_string
+        else:
+            # self._error_logger.error('Неверный формат даты')
+            raise ValueError('Неверный формат даты')
+        return val
+
+    # Проверка типа документа
+    def _check_document_type(self, val):
+        _valid_document_types = {
+            'Паспорт': ['Паспорт', 'паспорт'],
+            'Водительские права': ['Водительские права', 'водительские права', 'права'],
+            'Заграничный паспорт': ['Заграничный паспорт', 'заграничный паспорт', 'загран'],
+        }
+        found = False  # флаг найденного значения
+        for key, valid_list in _valid_document_types.items():
+            if val in valid_list:  # Проверка нахождения подаваемого значения в списке допустимых значений
+                val = key  # В случае успеха приведем значение к человеческому виду
+                found = True  # соответственно флаг теперь тру
+        if not found:  # Если подаваемого значения в списке нет, то исключение
+            # self._error_logger.error('Неверный тип документа')
+            raise ValueError('Неверный тип документа')
+        return val
+
+    # Проверка номера телефона
+    def _check_phone(self, val):
+        val = re.sub(r'[^0-9]+', r'', val)  # для начала удалим все лишнее(все то, что не цифра)
+        if len(val) != 11:  # количество цифр вместе с восьмеркой или семеркой
+            # self._error_logger.error('Неверное количество цифр в номере телефона')
+            raise ValueError('Неверное количество цифр в номере телефона')
+        return val
+
+    # Проверка номера документа
+    def _check_doc_id(self, val, obj):  # в doc_type будем передавать значение из dict объекта по ключу document_type
+        _valid_len = {
+            'Паспорт': 10,
+            'Водительские права': 10,
+            'Заграничный паспорт': 9,
+        }
+        doc_type = obj.__dict__.get('document_type')
+        v_len = _valid_len[doc_type]  # Получим валидную длину указанного(doc_type) документа
+        val = re.sub(r'[^0-9]+', r'', val)  # удалим все лишнее и непонятное
+
+        if len(val) != v_len:
+            # self._error_logger.error('Неверное количество цифр в номере документа')
+            raise ValueError('Неверное количество цифр в номере документа ')
+        return val
+
+    def _check_name(self, val, obj):
+        if not val.isalpha():
+            # self._error_logger.error(f'Невалидное {self.name} {val}')
+            raise ValueError(f'Невалидное {self.name} {val}')
+        if self.name in obj.__dict__:  # Если такое имя уже существует у объекта (позволяет избежать записи
+            # self._error_logger.error(f'Попытка изменить {self.name}')  # о смене фамилии или имени с None на текущую)
+            raise AttributeError(f'Попытка изменить {self.name}')
+        return val
+
+    def __init__(self, name='название_атрибута'):
+        # self._info_logger = logging.getLogger('info_logger')
+        # self._error_logger = logging.getLogger('error_logger')
+        self.name = name
+
+    def __get__(self, obj, objtype):
+        return obj.__dict__[self.name]  # вернем значение атрибута по ключу(названию атрибута)
+
+    @logging_decor
+    def __set__(self, obj, val):
+        if not isinstance(val, str):  #
+            # self._error_logger.error(f'Неверный тип {val} - {type(val)}')
+            raise TypeError(f'Неверный тип {val} - {type(val)}')
+
+        if self.name in ['first_name', 'last_name']:
+            val = self._check_name(val, obj)
+        elif self.name == 'birth_date':
+            val = self._check_date(val)
+        elif self.name == 'phone':
+            val = self._check_phone(val)
+        elif self.name == 'document_type':
+            val = self._check_document_type(val)
+        elif self.name == 'document_id':
+            val = self._check_doc_id(val, obj)  # передаем объект для получения типа документа
+
+        if self.name in obj.__dict__:  # изменяем только в том случае, когда объект существует
+            pass
+            # self._info_logger.info(
+            #     f"Изменяю {self.name} у пациента {obj.first_name} {obj.last_name} c {obj.__dict__.get(self.name)} на {val}")
+        obj.__dict__[self.name] = val
+
+
+class Patient:
+    first_name = LogAndChange('first_name')
+    last_name = LogAndChange('last_name')
+    birth_date = LogAndChange('birth_date')
+    phone = LogAndChange('phone')
+    document_type = LogAndChange('document_type')
+    document_id = LogAndChange('document_id')
+    current_row_idx = 1  # указатель на текущую строку
+
+    @logging_decor
+    def __init__(self, first_name, last_name, birth_date, phone, document_type, document_id):
+        # self._info_logger = logging.getLogger('info_logger')
+
+        if not (first_name and last_name and birth_date
+                and phone and document_type and document_id):
+            raise TypeError('Заполнены не все поля')
+
+        self.first_name = first_name
+        self.last_name = last_name
+        self.birth_date = birth_date
+        self.phone = phone
+        self.document_type = document_type
+        self.document_id = document_id
+
+        self._saved = False  # флаг, позволяющий узнать сохранен ли объект
+        # в начале объектов у нас нет
+        self._row_idx = None  # индекс объекта
+        # self._info_logger.info(f'Создан объект {self.first_name} {self.last_name} '
+        #                        f'{self.birth_date} {self.phone} {self.document_type} {self.document_id}')
+        #ретурн строка выше
+
+    @staticmethod
+    def create(first_name, last_name, birth_date, phone, document_type, document_id):
+        return Patient(first_name, last_name, birth_date, phone, document_type, document_id)
+
+        # Сохраняем пациента в csv таблицу
+
+    def save(self):
+        data = [self.first_name,
+                self.last_name,
+                self.birth_date,
+                self.phone,
+                self.document_type,
+                self.document_id]
+        if self._saved:  # Если пациент уже существует
+            #получаем его индекс cur.fetchall()[self._row_idx][0] и меняем данные пациента
+
+            #
+            # cur.execute("SELECT * FROM patients") #смотрим на пациентов
+            # cur.fetchall()[self._row_idx][0] # индекс пациента
+            # где-нибудь здесь получим индекс и сделаем UPDATE распакуем пациента в БД
+            cur.execute(f"UPDATE patients SET first_name = '{self.first_name}', last_name = '{self.last_name}',"
+                        f" birth_date = '{self.birth_date}',"
+                        f"phone = '{self.phone}', document_type = '{self.document_type}',"
+                        f" document_id = '{self.document_id}' WHERE patient_id = {self._row_idx}")
+            conn.commit()
+
+            # элементу списка со всеми пациентами присваются данные конкретного пациента
+
+        else:
+            self._saved = True  # Теперь пациент существует
+            self._row_idx = Patient.current_row_idx  # конкретном объекту присвается номер строки в файле
+            Patient.current_row_idx += 1  # после этого для следующего объекта счетчик +1
+
+            # БД covid
+            cur.execute("INSERT INTO patients (first_name, last_name, birth_date, "
+                        "phone, document_type, document_id) VALUES (%s, %s, %s, %s, %s, %s)",
+                        (self.first_name,
+                self.last_name,
+                self.birth_date,
+                self.phone,
+                self.document_type,
+                self.document_id)) # распакуем пациента
+            conn.commit()
+    def __str__(self):
+        return (f'{self.first_name} {self.last_name} {self.birth_date} '
+                f'{self.phone} {self.document_type} {self.document_id}')
+
+
+class PatientCollection:
+    # def __init__(self, path_to_file):
+    #     self.path_to_file = path_to_file
+
+
+    def __iter__(self):
+        #создать курсор, установить размер буфера на одну запись и итерироваться по курсору
+        cur.execute("SELECT * FROM patients ORDER BY patient_id")
+        for i in cur.fetchall():
+            # print(i)
+            row = [i[1], i[2], i[3], i[4], i[5], i[6]]
+            p = Patient(*row)
+            # p._saved = True  # пациент сохранен
+            # p._row_idx = i  # в строке i
+            yield p
+
+    def limit(self, num):
+        cur.execute("SELECT * FROM patients ORDER BY patient_id")
+        for i in cur.fetchall():
+            # print(i)
+            if i[0]> num:
+                break
+
+            row = [i[1], i[2], i[3], i[4], i[5], i[6]]
+            p = Patient(*row)
+            # p._saved = True  # пациент сохранен
+            # p._row_idx = i  # в строке i
+            yield p
+
+    # def __str__(self):
+    #     return (f'{Patient.first_name} {Patient.last_name} {Patient.birth_date} '
+    #             f'{Patient.phone} {Patient.document_type} {Patient.document_id}')
+
+
+if __name__ == "__main__":
+
+
+    # p = Patient('Vasya', 'Pupkin','08-04-1994','89126116381','паспорт','5162333666')
+    # p.save()
+    # p.birth_date = '12-10-1994'
+    # p.save()
+    #
+    # p = Patient('Pip', 'Lil','08-04-1995','89126116456','паспорт','5162444666')
+    # p.save()
+    # p = Patient('Lil', 'Nas','08-04-1984','89126456891','паспорт','5162555666')
+    # p.save()
+    #
+    p = Patient('kek', 'shpek', '08-04-1995', '89126116456', 'паспорт', '5162444666')
+    p.save()
+    p = Patient('mem', 'kekov', '08-04-1984', '89126456891', 'паспорт', '5162555666')
+    p.save()
+
+    p = Patient('Peton', 'ivanov', '08-04-1984', '89126456891', 'паспорт', '5162555666')
+    p.save()
+    p = Patient('Vasiliii', 'Petrov', '08-04-1984', '89126456891', 'паспорт', '5162555666')
+    p.save()
+    # p = Patient('Bill', 'Gates', '08-04-1984', '89126456891', 'паспорт', '51555666')
+    # p.save()
+
+    # pc = PatientCollection()
+    # for i in pc:
+    #     print(i)
+    # print()
+    #
+    #
+    # for x in pc.limit(5):
+    #     print(x)

README.md

@@ -1,10 +1,31 @@
-# python_developer_hw2
-Репозиторий для проверки домашки №2 по курсу "Разработчик Python"
+ Для работы необходимо в файле conf.py заменить данные на свои
+
+ Запуск из консоли: 
+
+Создание python cli.py create Имя Фамилия --birth-date 1990-01-01 --phone +7-916-000-00-00 --document-type паспорт --document-number 0000000000
 
-## Проверка ДЗ
+Вывод на экран первых 10 пользователей: python cli.py show
 
-1. В файле homework/config.py заполняете все константы своими значениями
-2. Размещаете классы Patient и PatientCollection в файле homework/patient.py
-3. ```pytest tests```
+Вывод на экран произвольного количества пользователей: python cli.py show 8
 
-Если всё ок, то Pull-request в этот репозиторий 
+Вывод на экран количества сохраненных пользователей: python cli.py count
+
+P.S.
+ Перед созданием пациента логируется изменение полей, как сделать в декораторе так, чтобы логирование шло именно при реальном изменении, пока не придумал.
+
+P.P.S так же необходимо проделать:
+
+pip3 install psycopg2-binary
+
+В Linux может потребоваться sudo apt install libpq-dev
+
+Ну и сам postgres:
+
+Create the file /etc/apt/sources.list.d/pgdg.list and add a line for the repository
+
+deb http://apt.postgresql.org/pub/repos/apt/ YOUR_UBUNTU_VERSION_HERE-pgdg main
+
+Import the repository signing key, and update the package lists
+
+wget --quiet -O - https://www.postgresql.org/media/keys/ACCC4CF8.asc | sudo apt-key add -
+sudo apt-get update

cli.py

@@ -0,0 +1,45 @@
+import click
+from HW3 import Patient, PatientCollection
+
+@click.group()
+def cli():
+    """Создаем пациентов через консоль"""
+@click.command('create')
+@click.argument('first-name')
+@click.argument('last-name')
+@click.option('--birth-date', help='The birth date format "yyyy-mm-dd"')
+@click.option('--phone', help='The phone format "8-ddd-ddd-dd-dd"')
+@click.option('--document-type', help='The document type must be "Паспорт", "Заграничный паспорт",'
+                                        ' "Водительские права"')
+@click.option('--document-number', help='The document id must have 10 digits for russian passport or driving licence'
+                                        'or 9 digits for international passport')
+
+def create_patient(first_name, last_name, birth_date, phone, document_type, document_number):
+    p = Patient(first_name, last_name, birth_date, phone, document_type, document_number)
+    p.save()
+
+
+@click.command('show')
+@click.argument('num', default=10)
+def show_patients(num):
+    pc = PatientCollection()
+    for m in pc.limit(num):
+        click.echo(m)
+
+
+@click.command('count')
+# @click.argument('num', default=10)
+def count_patients():
+    pc = PatientCollection()
+    count=0
+    for m in pc:
+        count+=1
+    click.echo(count)
+
+
+
+cli.add_command(create_patient)
+cli.add_command(show_patients)
+cli.add_command(count_patients)
+cli()
+

conf.py

@@ -0,0 +1,8 @@
+DATA_BASE = 'covid'
+LOGIN = 'aleksandrskiba'
+PASSWD = '31471'
+HST = 'localhost'
+NPORT = 5432
+
+
+