File: yencode_Encoder.py

package info (click to toggle)
python-yenc 0.4.0-10
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 388 kB
  • sloc: python: 572; ansic: 429; sh: 9; makefile: 6
file content (74 lines) | stat: -rwxr-xr-x 2,601 bytes parent folder | download | duplicates (3)
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
#!/usr/bin/env python
# -*- coding: utf-8 -*-
##=============================================================================
 #
 # Copyright (C) 2003, 2011 Alessandro Duca <alessandro.duca@gmail.com>
 #
 # This library is free software; you can redistribute it and/or
 # modify it under the terms of the GNU Lesser General Public
 # License as published by the Free Software Foundation; either
 # version 2.1 of the License, or (at your option) any later version.
 #
 # This library is distributed in the hope that it will be useful,
 # but WITHOUT ANY WARRANTY; without even the implied warranty of
 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 # Lesser General Public License for more details.
 #
 # You should have received a copy of the GNU Lesser General Public
 # License along with this library; if not, write to the Free Software
 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 #=============================================================================
 # 
##=============================================================================

import sys
import os
import os.path
import yenc
import getopt
from stat import *
from binascii import crc32

def main():
    try:
        opts, args = getopt.getopt(sys.argv[1:], "o:")
    except getopt.GetoptError:
        usage()
    try:  # Python 3
        file_out = sys.stdout.buffer
    except AttributeError:  # Python < 3
        file_out = sys.stdout
    for o,a in opts:
        if o == '-o':
            file_out = open(a,"wb")
    if args:
        filename = args[0]
        if not os.access( filename, os.F_OK | os.R_OK ):
            sys.stderr.write("couldn't access %s\n" % filename)
            sys.exit(2)
    else:
        usage()
    with open(filename,"rb") as file_in:
        crc = "%x"%(0xFFFFFFFF & crc32(file_in.read()))
    name = os.path.split(filename)[1]
    size = os.stat(filename)[ST_SIZE]
    line = "=ybegin line=128 size=%d crc32=%s name=%s\r\n"
    file_out.write((line % (size, crc, name)).encode("ascii"))
    with open(filename, "rb") as file_in:
        encoder = yenc.Encoder(file_out)
        while True:
            data_in = file_in.read(1024)
            encoder.feed(data_in)
            encoder.flush()
            if len(data_in) < 1024: break
    encoder.terminate()
    encoder.flush()
    line = "=yend size=%d crc32=%s\r\n" % (size, encoder.getCrc32())
    file_out.write(line.encode("ascii"))

def usage():
    sys.stderr.write("Usage: yencode_Encoder.py <-o outfile> filename\n")
    sys.exit(1)

if __name__ == "__main__":
    main()