ncrsc/upload.py at master · Ralfk2/ncrsc · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
import pyrs.comms
import pyrs.rpc
import pyrs.msgs
import curses

# Message Definitions.
from pyrs.proto import core_pb2
from pyrs.proto import system_pb2
from pyrs.proto import files_pb2

class Menu(object):
    def __init__(self, nc_window, rs_rpc, parser, parent):
        #menu - structure
        self.title = "Uploads"
        self.cur_funct = self.print_active_uploads

        my,mx = nc_window.getmaxyx()
        self.__nc_window = curses.newwin(my-9, mx-4, 8, 2); self.__nc_window.border(0); self.__nc_window.idlok(1)
        self.my,self.mx = self.__nc_window.getmaxyx()
        self.__rs_rpc = rs_rpc
        self.__parser = parser
        self.parent = parent
        self.scroll = 0
        self.transfers = False
        #self.chat_register_id = self.__rs_rpc.request(pyrs.msgs.constructMsgId(core_pb2.CORE, core_pb2.CHAT, chat_pb2.MsgId_RequestRegisterEvents, False), self.__rp)
    def tick(self, update=False):
        self.cur_funct(update)
    def end(self):
        return True
    def menu_key(self, c):
        if c == curses.KEY_DOWN:
            self.scroll += 5
        elif c == curses.KEY_UP:
            self.scroll -= 5
            if self.scroll < 0: self.scroll = 0
        return True
    def print_active_uploads(self, update=False):
        if update: self.transfers = self.list_downloads()
        if self.transfers:
            self.__nc_window.erase()
            i=1
            for transfer in self.transfers:
                if i > self.scroll:
                    if i+2 < self.my:
                        self.__nc_window.addstr(1+i-self.scroll, 1, str(i)+": name: "+transfer.file.name.encode("utf-8"))
                        self.__nc_window.move(1+i-self.scroll, int(self.mx/3.)+5); self.__nc_window.clrtoeol()
                        self.__nc_window.addstr(1+i-self.scroll, int(self.mx/3.)+5, "size: "+str(transfer.file.size/1024)+" kB")
                        self.__nc_window.move(1+i-self.scroll, int(self.mx/2.)); self.__nc_window.clrtoeol()
                        self.__nc_window.addstr(1+i-self.scroll, int(self.mx/2.), str(round(transfer.fraction*100,2))+"% finished")
                        self.__nc_window.move(1+i-self.scroll, int(2*self.mx/3.)); self.__nc_window.clrtoeol()
                        self.__nc_window.addstr(1+i-self.scroll, int(2*self.mx/3.), "rate: "+str(round(transfer.rate_kBs,2))+" kB/s")
                i+=1
            self.__nc_window.addstr(1, 1, "active Uploads ("+str(i-1)+"):")
            if self.scroll > 0 and i+2-self.scroll < self.my:
                self.scroll -= self.my-(i+2-self.scroll)
                if self.scroll < 0: self.scroll = 0
        else:
            self.__nc_window.erase()
            self.__nc_window.addstr(1, 1, "no active Uploads")
        self.__nc_window.border(0)
        self.__nc_window.refresh()
    def list_downloads(self):
        rp = files_pb2.RequestTransferList()
        rp.direction = files_pb2.DIRECTION_UPLOAD
        msg_id = pyrs.msgs.constructMsgId(core_pb2.CORE, core_pb2.FILES, files_pb2.MsgId_RequestTransferList, False);
        req_id = self.__rs_rpc.request(msg_id, rp)
        (msg_id, msg_body) = self.parent.rs_response(req_id)
        if msg_body :
            resp = self.__parser.construct(msg_id, msg_body)
            if resp.status.code == core_pb2.Status.SUCCESS:
                return  resp.transfers
            else : return False
    def get_help(self):
        help_list = ["   move up/down with your arrow keys"]
        return help_list