def save_transferred(self, raw_fname, transferred, output_dir=None):
output_fpath = "transferred_{0}.txt".format(
os.path.splitext(os.path.basename(raw_fname))[0])
if output_dir is not None:
if not os.path.isdir(output_dir):
os.makedirs(output_dir)
output_fpath = os.path.join(output_dir, output_fpath)
return (Doraemon.put_file_contents(output_fpath,
transferred), output_fpath)
def test_put_file_contents_str(self):
s, fpath = "123", os.path.join(self.resources, "foo.bar")
self.assertTrue(not os.path.isfile(fpath))
self.assertTrue(Doraemon.put_file_contents(fpath, s))
self.assertTrue(os.path.isfile(fpath))
self.assertEqual(1, len(Doraemon.get_file_contents(fpath)))
self.assertEqual(s, Doraemon.get_file_contents(fpath)[0])
if os.path.isfile(fpath):
os.remove(fpath)
self.assertTrue(not os.path.isfile(fpath))
def test_put_file_contents_list(self):
_l, fpath = range(0, 10), os.path.join(self.resources, "foo.bar")
self.assertTrue(not os.path.isfile(fpath))
self.assertTrue(Doraemon.put_file_contents(fpath, _l))
self.assertTrue(os.path.isfile(fpath))
self.assertEqual(10, len(Doraemon.get_file_contents(fpath)))
self.assertEqual(map(lambda num: str(num), _l),
Doraemon.get_file_contents(fpath))
if os.path.isfile(fpath):
os.remove(fpath)
self.assertTrue(not os.path.isfile(fpath))
# coding=utf-8
import os
import sys
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
if __name__ == "__main__":
from doraemon import Doraemon
fpath = sys.argv[1]
contents = Doraemon.get_file_contents(fpath,
in_charset="gbk",
out_charset="utf-8")
Doraemon.put_file_contents("utf_8_{0}".format(os.path.basename(fpath)),
contents)