mirror of
https://github.com/paul-nameless/tg
synced 2024-11-22 20:03:17 +00:00
Merge pull request #2 from paul-nameless/remove-controller-duplicated-file
remove 2nd controller & update gitignore
This commit is contained in:
Nameless
GitHub
commit
dbd3b29b50
2 changed files with 1 additions and 162 deletions
1
.gitignore
vendored
1
.gitignore
vendored
|
|
@ -1,3 +1,4 @@
|
||||||
|
.mypy_cache/
|
||||||
__pycache__
|
__pycache__
|
||||||
.env
|
.env
|
||||||
dist
|
dist
|
||||||
|
|
|
||||||
162
controller.py
162
controller.py
|
|
@ -1,162 +0,0 @@
|
||||||
import logging
|
|
||||||
import os
|
|
||||||
import threading
|
|
||||||
|
|
||||||
from utils import notify
|
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
|
||||||
|
|
||||||
|
|
||||||
class Controller:
|
|
||||||
"""
|
|
||||||
# MVC
|
|
||||||
# Model is data from telegram
|
|
||||||
# Controller handles keyboad events
|
|
||||||
# View is terminal vindow
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(self, model, view):
|
|
||||||
self.model = model
|
|
||||||
self.view = view
|
|
||||||
self.lock = threading.Lock()
|
|
||||||
|
|
||||||
def run(self):
|
|
||||||
try:
|
|
||||||
self.handle_chats()
|
|
||||||
except Exception as e:
|
|
||||||
logger.exception('Error happened in main loop')
|
|
||||||
|
|
||||||
def handle_msgs(self):
|
|
||||||
# set width to 0.25, move window to left
|
|
||||||
# refresh everything
|
|
||||||
self.view.chats.resize(0.2)
|
|
||||||
self.view.msgs.resize(0.2)
|
|
||||||
self.refresh_chats()
|
|
||||||
|
|
||||||
while True:
|
|
||||||
|
|
||||||
key = self.view.get_key(self.view.chats.h, self.view.chats.w)
|
|
||||||
logger.info('Pressed key: %s', key)
|
|
||||||
if key == 'q':
|
|
||||||
return 'QUIT'
|
|
||||||
elif key == ']':
|
|
||||||
if self.model.next_chat():
|
|
||||||
self.refresh_chats()
|
|
||||||
elif key == '[':
|
|
||||||
if self.model.prev_chat():
|
|
||||||
self.refresh_chats()
|
|
||||||
elif key == 'J':
|
|
||||||
if self.model.jump_next_msg():
|
|
||||||
self.refresh_msgs()
|
|
||||||
elif key == 'K':
|
|
||||||
if self.model.jump_prev_msg():
|
|
||||||
self.refresh_msgs()
|
|
||||||
elif key in ('j', '^B'):
|
|
||||||
if self.model.next_msg():
|
|
||||||
self.refresh_msgs()
|
|
||||||
elif key in ('k', '^C'):
|
|
||||||
if self.model.prev_msg():
|
|
||||||
self.refresh_msgs()
|
|
||||||
elif key == 'G':
|
|
||||||
if self.model.jump_bottom():
|
|
||||||
self.refresh_msgs()
|
|
||||||
|
|
||||||
elif key == '/':
|
|
||||||
# search
|
|
||||||
pass
|
|
||||||
elif key == 'gg':
|
|
||||||
# move to the top
|
|
||||||
pass
|
|
||||||
elif key == 'e':
|
|
||||||
# edit msg
|
|
||||||
pass
|
|
||||||
elif key == 'r':
|
|
||||||
# reply to this msg
|
|
||||||
# print to status line
|
|
||||||
pass
|
|
||||||
elif key == 'I':
|
|
||||||
# open vim or emacs to write long messages
|
|
||||||
pass
|
|
||||||
elif key == 'i':
|
|
||||||
# write new message
|
|
||||||
msg = self.view.get_input()
|
|
||||||
if msg:
|
|
||||||
chat_id = self.model.get_current_chat_id()
|
|
||||||
self.model.msgs.tg.send_message(
|
|
||||||
chat_id=chat_id,
|
|
||||||
text=msg,
|
|
||||||
)
|
|
||||||
self.view.draw_status(f'Sent: {msg}')
|
|
||||||
|
|
||||||
elif key in ('h', '^D'):
|
|
||||||
return 'BACK'
|
|
||||||
|
|
||||||
def handle_chats(self):
|
|
||||||
# set width to 0.5, move window to center?
|
|
||||||
# refresh everything
|
|
||||||
self.view.chats.resize(0.5)
|
|
||||||
self.view.msgs.resize(0.5)
|
|
||||||
self.refresh_chats()
|
|
||||||
while True:
|
|
||||||
|
|
||||||
key = self.view.get_key(self.view.chats.h, self.view.chats.w)
|
|
||||||
logger.info('Pressed key: %s', key)
|
|
||||||
if key == 'q':
|
|
||||||
return
|
|
||||||
elif key in ('l', '^E'):
|
|
||||||
rc = self.handle_msgs()
|
|
||||||
if rc == 'QUIT':
|
|
||||||
return
|
|
||||||
self.view.chats.resize(0.5)
|
|
||||||
self.view.msgs.resize(0.5)
|
|
||||||
self.refresh_chats()
|
|
||||||
|
|
||||||
elif key in ('j', '^B'):
|
|
||||||
is_changed = self.model.next_chat()
|
|
||||||
if is_changed:
|
|
||||||
self.refresh_chats()
|
|
||||||
|
|
||||||
elif key in ('k', '^C'):
|
|
||||||
is_changed = self.model.prev_chat()
|
|
||||||
if is_changed:
|
|
||||||
self.refresh_chats()
|
|
||||||
|
|
||||||
def refresh_chats(self):
|
|
||||||
self.view.draw_chats(
|
|
||||||
self.model.current_chat,
|
|
||||||
self.model.get_chats(limit=self.view.chats.h)
|
|
||||||
)
|
|
||||||
self.refresh_msgs()
|
|
||||||
self.view.draw_status()
|
|
||||||
|
|
||||||
def refresh_msgs(self):
|
|
||||||
self.view.msgs.users = self.model.users
|
|
||||||
msgs = self.model.get_current_msgs(limit=self.view.msgs.h)
|
|
||||||
self.view.draw_msgs(self.model.get_current_msg(), msgs)
|
|
||||||
|
|
||||||
def update_handler(self, update):
|
|
||||||
logger.debug('===============Received: %s', update)
|
|
||||||
_type = update['@type']
|
|
||||||
if _type == 'updateNewMessage':
|
|
||||||
logger.debug('Updating... new message')
|
|
||||||
# with self.lock:
|
|
||||||
chat_id = update['message']['chat_id']
|
|
||||||
self.model.msgs.msgs[chat_id].append(update['message'])
|
|
||||||
# msgs = self.model.get_current_msgs()
|
|
||||||
self.refresh_msgs()
|
|
||||||
if not update['disable_notification']:
|
|
||||||
try:
|
|
||||||
notify(update['message']['content']['text']['text'])
|
|
||||||
except Exception:
|
|
||||||
logger.exception('Error happened on notify: %s', update)
|
|
||||||
# message_content = update['message']['content'].get('text', {})
|
|
||||||
# we need this because of different message types: photos, files, etc.
|
|
||||||
# message_text = message_content.get('text', '').lower()
|
|
||||||
|
|
||||||
# if message_text == 'ping':
|
|
||||||
# chat_id = update['message']['chat_id']
|
|
||||||
# # print(f'Ping has been received from {chat_id}')
|
|
||||||
# self.tg.send_message(
|
|
||||||
# chat_id=chat_id,
|
|
||||||
# text='pong',
|
|
||||||
# )
|
|
||||||
Loading…
Reference in a new issue