~jbauer/cloud-init/salt

« back to all changes in this revision

Viewing changes to cloudinit/SshUtil.py

  • Committer: Scott Moser
  • Date: 2012-01-17 20:59:21 UTC
  • mfrom: (505.1.4 trunk.cleanups)
  • Revision ID: smoser@ubuntu.com-20120117205921-pyf7ybik03gq3t3g
miscellaneous cleanups, and add tools/run-pylint

adding run-pylint makes it easy to run pylint with given configuration
against the code.

Show diffs side-by-side

added added

removed removed

Lines of Context:
156
156
    os.umask(saved_umask)
157
157
 
158
158
 
159
 
if __name__ == "__main__":
160
 
    import sys
161
 
    # usage: orig_file, new_keys, [key_prefix]
162
 
    #   prints out merged, where 'new_keys' will trump old
163
 
    ##  example
164
 
    ## ### begin auth_keys ###
165
 
    # ssh-rsa AAAAB3NzaC1xxxxxxxxxV3csgm8cJn7UveKHkYjJp8= smoser-work
166
 
    # ssh-rsa AAAAB3NzaC1xxxxxxxxxCmXp5Kt5/82cD/VN3NtHw== smoser@brickies
167
 
    # ### end authorized_keys ###
168
 
    #
169
 
    # ### begin new_keys ###
170
 
    # ssh-rsa nonmatch smoser@newhost
171
 
    # ssh-rsa AAAAB3NzaC1xxxxxxxxxV3csgm8cJn7UveKHkYjJp8= new_comment
172
 
    # ### end new_keys ###
173
 
    #
174
 
    # Then run as:
175
 
    #  program auth_keys new_keys \
176
 
    #      'no-port-forwarding,command=\"echo hi world;\"'
177
 
    def_prefix = None
178
 
    orig_key_file = sys.argv[1]
179
 
    new_key_file = sys.argv[2]
180
 
    if len(sys.argv) > 3:
181
 
        def_prefix = sys.argv[3]
182
 
    fp = open(new_key_file)
183
 
 
184
 
    newkeys = []
185
 
    for line in fp.readlines():
186
 
        newkeys.append(AuthKeyEntry(line, def_prefix))
187
 
 
188
 
    fp.close()
189
 
    print update_authorized_keys(orig_key_file, newkeys)
190
 
 
191
 
 
192
159
def parse_ssh_config(fname="/etc/ssh/sshd_config"):
193
160
    ret = {}
194
161
    fp = open(fname)
200
167
        ret[key] = val
201
168
    fp.close()
202
169
    return(ret)
 
170
 
 
171
if __name__ == "__main__":
 
172
    def main():
 
173
        import sys
 
174
        # usage: orig_file, new_keys, [key_prefix]
 
175
        #   prints out merged, where 'new_keys' will trump old
 
176
        ##  example
 
177
        ## ### begin auth_keys ###
 
178
        # ssh-rsa AAAAB3NzaC1xxxxxxxxxV3csgm8cJn7UveKHkYjJp8= smoser-work
 
179
        # ssh-rsa AAAAB3NzaC1xxxxxxxxxCmXp5Kt5/82cD/VN3NtHw== smoser@brickies
 
180
        # ### end authorized_keys ###
 
181
        #
 
182
        # ### begin new_keys ###
 
183
        # ssh-rsa nonmatch smoser@newhost
 
184
        # ssh-rsa AAAAB3NzaC1xxxxxxxxxV3csgm8cJn7UveKHkYjJp8= new_comment
 
185
        # ### end new_keys ###
 
186
        #
 
187
        # Then run as:
 
188
        #  program auth_keys new_keys \
 
189
        #      'no-port-forwarding,command=\"echo hi world;\"'
 
190
        def_prefix = None
 
191
        orig_key_file = sys.argv[1]
 
192
        new_key_file = sys.argv[2]
 
193
        if len(sys.argv) > 3:
 
194
            def_prefix = sys.argv[3]
 
195
        fp = open(new_key_file)
 
196
 
 
197
        newkeys = []
 
198
        for line in fp.readlines():
 
199
            newkeys.append(AuthKeyEntry(line, def_prefix))
 
200
 
 
201
        fp.close()
 
202
        print update_authorized_keys(orig_key_file, newkeys)
 
203
 
 
204
    main()
 
205
 
 
206
# vi: ts=4 expandtab